Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
VisualGLM-6B_pytorch
Commits
d03b4039
Commit
d03b4039
authored
Jan 22, 2024
by
wangsen
Browse files
Update README.md
parent
ab6fb800
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
18 deletions
+1
-18
README.md
README.md
+1
-18
No files found.
README.md
View file @
d03b4039
...
...
@@ -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安装依赖
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment