Unverified 提交 7f8fcc81 作者: cocomany 提交者: GitHub

supports GPU usage in docker (#197)

上级 b95ad4fc
...@@ -46,7 +46,7 @@ sudo systemctl restart docker ...@@ -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
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论