Update README.md

README.md

@@ -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 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（方法二）
 
@@ -74,6 +75,7 @@ python3 prep_enwik8.py
 ## 训练
 ### 单机单卡
 ```
+cd pytorch
 sh run_enwik8_base.sh train
 ```
 ### 单机多卡