Updata dtk to 24.04

docker/Dockerfile

-FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest
-RUN source /opt/dtk/env.sh
-COPY requirements.txt requirements.txt
-RUN pip3 install -r requirements.txt
+FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310

train.sh

@@ -8,7 +8,6 @@ export MIOPEN_FIND_MODE=4
 
 echo "Training start ..."
 # coco_path是训练数据集地址，数据是coco format
-python -m torch.distributed.launch --nproc_per_node=4 --use_env main.py --coco_path /path/to/coco
-# python -m torch.distributed.launch --nproc_per_node=4 --use_env main.py --coco_path /home/datasets/COCO2017
+torchrun --nproc_per_node=4 --use_env main.py --coco_path /path/to/coco
 
 echo "Training finished."
\ No newline at end of file