Commit d03b4039 authored by wangsen's avatar wangsen
Browse files

Update README.md

parent ab6fb800
......@@ -18,7 +18,7 @@ VisualGLM-6B 由 SwissArmyTransformer(简称sat) 库训练,这是一个支持T
在光源可拉取推理的docker镜像,拉取方式如下:
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:visualglm-6b_pytorch-latest
docker run -it -v /path/your_data/:/path/your_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
```
### Dockerfile(方法二)
......@@ -27,23 +27,6 @@ docker run -it -v /path/your_data/:/path/your_data/ --shm-size=32G --privileged=
docker build --no-cache -t xxx:latest .
docker run -it -v /path/your_data/:/path/your_data/ --shm-size=32G --privileged=true --device=/dev/kfd --device=/dev/dri/ --group-add video --name docker_name imageID bash
```
### Anaconda(方法三)
此处提供本地配置、编译的详细步骤,例如:
关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
```
DTK驱动:dtk23.10
python:python3.8
torch:1.13.1
torchvision:0.14.1
deepspeed:0.12.3
```
其它非深度学习库参照requirements.txt安装:
```
pip install -r requirements.txt
```
## 推理
使用pip安装依赖
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment