瀏覽代碼

supports GPU usage in docker (#197)

cocomany 2 年之前
父節點
當前提交
7f8fcc81c4
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      README.md

+ 1 - 1
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