Commit c1bffbea authored by “yuguo”'s avatar “yuguo”
Browse files

update

parent 5796603c
......@@ -63,7 +63,7 @@ class Linear1D(nn.Module):
### Docker
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练以及推理的docker镜像:image.sourcefind.cn:5000/dcu/admin/base/oneflow:0.9.1-centos7.6-dtk-22.10.1-py39-latest
提供[光源](https://www.sourcefind.cn/#/service-details)拉取的训练以及推理的docker镜像:image.sourcefind.cn:5000/dcu/admin/base/oneflow:0.9.1-centos7.6-dtk-22.10.1-py39-latest,关于本项目DCU显卡所需torch库等均可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装
docker pull image.sourcefind.cn:5000/dcu/admin/base/oneflow:0.9.1-centos7.6-dtk-22.10.1-py39-latest
# <Your Image ID>用上面拉取docker镜像的ID替换
......@@ -72,6 +72,7 @@ class Linear1D(nn.Module):
pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple
pip3 install pybind11 -i https://mirrors.aliyun.com/pypi/simple
pip3 install -e . -i https://mirrors.aliyun.com/pypi/simple
pip3 install torch-1.10.0a0+git2040069.dtk2210-cp39-cp39-manylinux2014_x86_64.whl
## 数据集
......@@ -174,7 +175,7 @@ demo.py如下:
if dist.is_main_process():
print(res)
### 输出
### result
```
>>>Total number of model parameters: 9,879,633,920
......@@ -187,7 +188,7 @@ demo.py如下:
将模型权重放置与glm-QA.py同一目录下,运行以下代码:
cd projects/GLM`
cd projects/GLM
python3 -m oneflow.distributed.launch --nproc_per_node 4 glm-QA.py
程序运行起来后,允许用户在命令行进行问答交互,输入“退出”,可以结束程序,如下所示:
......
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