Skip to content

GitLab

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

update

parent 5796603c
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 3 deletions
README.md
View file @ c1bffbea
...@@ -63,7 +63,7 @@ class Linear1D(nn.Module): ...@@ -63,7 +63,7 @@ class Linear1D(nn.Module):
### Docker ### 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 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替换 # <Your Image ID>用上面拉取docker镜像的ID替换
...@@ -72,6 +72,7 @@ class Linear1D(nn.Module): ...@@ -72,6 +72,7 @@ class Linear1D(nn.Module):
pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple
pip3 install pybind11 -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 -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如下： ...@@ -174,7 +175,7 @@ demo.py如下：
if dist.is_main_process(): if dist.is_main_process():
print(res) print(res)
### 输出 ### result
``` ```
>>>Total number of model parameters: 9,879,633,920 >>>Total number of model parameters: 9,879,633,920
...@@ -187,7 +188,7 @@ demo.py如下： ...@@ -187,7 +188,7 @@ demo.py如下：
将模型权重放置与glm-QA.py同一目录下，运行以下代码： 将模型权重放置与glm-QA.py同一目录下，运行以下代码：
cd projects/GLM` cd projects/GLM
python3 -m oneflow.distributed.launch --nproc_per_node 4 glm-QA.py 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