search

encounter1997 / DE-DETRs

Official Implementation of DE-DETR and DELA-DETR in "Towards Data-Efficient Detection Transformers"
Apache License 2.0
79 stars 8 forks source link

one server has some GPUs，I choose use one GPU,but the code did not run successfully #2

Open InterstellarFang opened 2 years ago

InterstellarFang commented 2 years ago

Thank you for your contribution! Enter on the command line: CUDA_VISIBLE_DEVICES=1 python -m torch.distributed.launch --nproc_per_node=1 --master_port=29501 --use_env main.py --dataset_file cityscapes --coco_path data/cityscapes --batch_size 4 --model de-detr --wandb

The problem is that the code still runs on GPU 0,GPU 0 is used by other student(RuntimeError: CUDA out of memory.Tried to allocate MiB.),I want run code on GPU 1,can you give me some help?Thank you very much!