diff --git a/README.md b/README.md index 7c4a8a9..c42da6a 100644 --- a/README.md +++ b/README.md @@ -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