How to train with Multi-GPUs

daitao / SAN

Second-order Attention Network for Single Image Super-resolution (CVPR-2019)

550 stars 106 forks source link

How to train with Multi-GPUs #26

Open robinaman opened 4 years ago

robinaman commented 4 years ago

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

Akshaypatil7 commented 3 years ago

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

dkjangid910ucsb commented 3 years ago

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 " https://github.com/pytorch/pytorch/issues/8637". It is written as testModule as given in the first answer.