"super-resolution.ipynb" , multiple GPU support?

[mohammad-saber]

Hi,

Thanks a lot for sharing your great work. I am using "super-resolution.ipynb". I came up with several questions in this regard. I submit them one-by-one. If you have any time, I really appreciate it to help me.

I found that "super-resolution.ipynb" code uses only one GPU (GPU number 0). Even though, I added the following code at the top:

os.environ['CUDA_VISIBLE_DEVICES'] = '0, 1, 2'

Still, only one GPU was running.

Is it possible that I can use data parallel to support multiple GPU?

 model = nn.DataParallel(model)