From 7f8fcc81c4f0bd41c67da147a9ab07bc4ca8a05f Mon Sep 17 00:00:00 2001
From: cocomany <124849750+cocomany@users.noreply.github.com>
Date: Sat, 29 Apr 2023 22:40:35 +0800
Subject: [PATCH] supports GPU usage in docker (#197)

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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