search

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
MIT License
2.43k stars 669 forks source link

multi-GPU #196

Closed DanChen001 closed 5 years ago

DanChen001 commented 5 years ago

Thank you for sharing the code.

How to trian this network with multi-GPU? how to set the option？ Thanks

sanghyun-son commented 5 years ago

Hello,

You can use the --n_GPUs argument for a multi-GPU training.

Ex)

python main.py [your arguments] --n_GPUs 4

There can be some instabilities or minor bugs, so just let me know if you face those problems.

Thank you!

juingzhou commented 5 years ago

Hello, I used multiple GPUs with the following errors RuntimeError: Expected tensor for argument #1 'input' to have the same device as tensor for argument #2 'weight'; but device 1 does not equal 0 (while checking arguments for cudnn_convolution)

Akshaypatil7 commented 4 years ago

@juingzhou have you solved this issue