Skip to content

GitLab

Commit 35d29663 authored by Rayyyyy's avatar Rayyyyy
Browse files

Updata dtk to 24.04

parent fb8fe5f4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 6 deletions
docker/Dockerfile
View file @ 35d29663
FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:1.13.1-centos7.6-dtk-23.04.1-py38-latest FROM image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk24.04-py310
RUN source /opt/dtk/env.sh
COPY requirements.txt requirements.txt
RUN pip3 install -r requirements.txt
train.sh
View file @ 35d29663
...@@ -8,7 +8,6 @@ export MIOPEN_FIND_MODE=4 ...@@ -8,7 +8,6 @@ export MIOPEN_FIND_MODE=4
echo "Training start ..." echo "Training start ..."
# coco_path是训练数据集地址，数据是coco format # coco_path是训练数据集地址，数据是coco format
python -m torch.distributed.launch --nproc_per_node=4 --use_env main.py --coco_path /path/to/coco torchrun --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
echo "Training finished." echo "Training finished."
\ No newline at end of file
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