How to train with Multi-GPUs

[robinaman]

according to the "option.py". it's ok to use "--n_GPUs". Howerver, it didn't work

[Akshaypatil7]

same question. Did you solved it. I also tried setting the : os.environ["CUDA_VISIBLE_DEVICES"] = "0,1"

[dkjangid910ucsb]

SAN network does not work on Multiple GPUs. I have written codes for other models and it works fine on multiple GPUs. I figured out that following may be the reason for the SAN network " pytorch/pytorch#8637". It is written as testModule as given in the first answer.