Skip to content

GitLab

"tests/backend/vscode:/vscode.git/clone" did not exist on "c24e285a8c57c17f38e6bda240ba6468483c7614"
Commit cb2aecce authored by dcuai's avatar dcuai
Browse files

Update README.md

parent 610585d3
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
README.md
View file @ cb2aecce
...@@ -24,6 +24,7 @@ transformer-XL ...@@ -24,6 +24,7 @@ transformer-XL
docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest
docker run -it --network=host --name=transformer-XL --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=32G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest docker run -it --network=host --name=transformer-XL --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=32G --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest
pip install -r requirements.txt
``` ```
### Dockerfile（方法二） ### Dockerfile（方法二）
...@@ -74,6 +75,7 @@ python3 prep_enwik8.py ...@@ -74,6 +75,7 @@ python3 prep_enwik8.py
## 训练 ## 训练
### 单机单卡 ### 单机单卡
``` ```
cd pytorch
sh run_enwik8_base.sh train sh run_enwik8_base.sh train
``` ```
### 单机多卡 ### 单机多卡
......
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