supports GPU usage in docker (#197)

This commit is contained in:
cocomany 2023-04-29 22:40:35 +08:00 committed by GitHub
parent b95ad4fcbc
commit 7f8fcc81c4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -46,7 +46,7 @@ sudo systemctl restart docker
```
安装完成后,可以使用以下命令编译镜像和启动容器:
```
docker build -t chatglm-cuda:latest .
docker build -f Dockerfile-cuda -t chatglm-cuda:latest .
docker run --gpus all -d --name chatglm -p 7860:7860 chatglm-cuda:latest
#若要使用离线模型请配置好模型路径然后此repo挂载到Container