update README.md

Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.10.0-centos7.6-dtk-22.10-py37-latest
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 
 COPY requirements.txt requirements.txt
-RUN source /opt/dtk-22.10/env.sh
+RUN source /opt/dtk/env.sh
 RUN cp /usr/share/zoneinfo/Asia/Shanghai /etc/localtime && echo 'Asia/Shanghai' >/etc/timezone 
 ENV LANG C.UTF-8
 RUN pip install -r requirements.txt -i http://mirrors.aliyun.com/pypi/simple/ --trusted-host mirrors.aliyun.com

README.md

@@ -22,8 +22,8 @@
 此处提供[光源](https://sourcefind.cn/#/main-page)拉取docker镜像
 
 ```
-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=flavr --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 pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10 
+docker run -it --network=host --name=flavr --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 -v /opt/hyhal:/opt/hyhal:ro --ulimit memlock=-1:-1 image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-ubuntu20.04-dtk24.04.1-py3.10
 pip install -r requirements.txt
 ```
 
@@ -33,7 +33,7 @@ dockerfile使用方法
 
 ```
 docker build --no-cache -t flavr:latest .
-docker run -dit --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G  --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 flavr:latest
+docker run -dit --network=host --name=flavr --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=16G  --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root -v /opt/hyhal:/opt/hyhal:ro --ulimit stack=-1:-1 --ulimit memlock=-1:-1 flavr:latest
 docker exec -it flavr /bin/bash
 ```
 
@@ -42,10 +42,10 @@ docker exec -it flavr /bin/bash
 关于本项目DCU显卡所需的特殊深度学习库可从[光合](https://developer.hpccube.com/tool/)开发者社区下载安装。
 
 ```
-DTK驱动：dtk22.10
-python：python3.7
-torch==1.10.0a0+git2040069.dtk2210
-torchvision==0.10.0a0+e04d001.dtk2210
+DTK驱动：dtk24.04.1
+python：python3.10
+torch==2.1.0+das1.1.git3ac1bdd.abi1.dtk2404
+torchvision==0.16.0+das1.1.git7d45932.abi1.dtk2404.torch2.1
 ```
 
 `Tips：以上dtk驱动、python等DCU相关工具版本需要严格一一对应`
@@ -83,6 +83,7 @@ pip install tensorboard setuptools==57.5.0 six
 ### 单机单卡
 
 ```
+注意：在k100_AI上运行需要在命令行前设置HIP_VISIBLE_DEVICES=xx，指定一下使用显卡
 python main.py --batch_size 32 --test_batch_size 32 --dataset vimeo90K_septuplet --loss 1*L1 --max_epoch 200 --lr 0.0002 --data_root datasets --n_outputs 1 --num_gpu 1
 ```
 

requirements.txt

@@ -2,3 +2,4 @@ imageio==2.31.2
 numpy
 Pillow==9.5.0
 tqdm==4.65.0
+tensorboard
\ No newline at end of file