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!
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!