Update README.md

README.md

@@ -25,7 +25,7 @@
 
 ```
 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=bert_prof --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=pix2pixhd --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
 ```
 
 ### Dockerfile（方法二）
@@ -48,6 +48,7 @@ DTK驱动：dtk22.10
 python：python3.7
 apex==0.1+gitdb7007a.dtk2210
 torch==1.10.0a0+git2040069.dtk2210
+torchvision==0.10.0a0+e04d001.dtk2210
 ```
 
 `Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
@@ -56,6 +57,7 @@ torch==1.10.0a0+git2040069.dtk2210
 
 ```
 pip install -r requirements.txt
+pip install numpy pillow==9.5.0 scipy
 ```
 
 ## 数据集
@@ -146,7 +148,7 @@ python train.py --name label2city_512p --batchSize 4 --gpu_ids 0,1,2,3
 
 ## 源码仓库及问题反馈
 
-* [https://github.com/NVIDIA/pix2pixHD](https://github.com/NVIDIA/pix2pixHD)
+* [https://developer.hpccube.com/codes/modelzoo/pix2pixhd_pytorch](https://developer.hpccube.com/codes/modelzoo/pix2pixhd_pytorch)
 ## 参考资料
 * [https://github.com/NVIDIA/pix2pixHD](https://github.com/NVIDIA/pix2pixHD)