soapisnotfat / super-resolution

collection of super-resolution models & algorithms
Apache License 2.0
744 stars 188 forks source link

About cuda #23

Closed quankainwafu closed 4 years ago

quankainwafu commented 5 years ago

I run 'python3 main -m 'srcnn'',it seems i just use cpu to train .how can i use gpu ? Thank you for your help.

Mondayyyy commented 5 years ago

I have the same problem. How to resolve it? Thanks you for your help.