The problem of two gpus training
cici-999 opened this issue · 2 comments
cici-999 commented
Hello, I want to train the model with two gpus, I changed os.environ['CUDA_VISIBLE_DEVICES']='0,1' and gpus=2, there is a RuntimeError:Distributed package doesn't have NCCL built in problem, please have any suggestions?
wondervictor commented
Hi @cici-999, how about trying CUDA_VISIBLE_DEVICES=0,1 python tools/train_net.py ...