supports GPU usage in docker (#197)
This commit is contained in:
parent
b95ad4fcbc
commit
7f8fcc81c4
|
|
@ -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
|
docker run --gpus all -d --name chatglm -p 7860:7860 chatglm-cuda:latest
|
||||||
|
|
||||||
#若要使用离线模型,请配置好模型路径,然后此repo挂载到Container
|
#若要使用离线模型,请配置好模型路径,然后此repo挂载到Container
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue