support GPU accelerated

satoshiiizuka / siggraph2016_colorization

Code for the paper 'Let there be Color!: Joint End-to-end Learning of Global and Local Image Priors for Automatic Image Colorization with Simultaneous Classification'.

http://hi.cs.waseda.ac.jp/~iizuka/projects/colorization/

Other

2.16k stars 302 forks source link

support GPU accelerated #22

Open sujameslin opened 6 years ago

sujameslin commented 6 years ago

Would you please teach me how to leverage GPU to run this code?

Thanks

Dawars commented 5 months ago

Might be late but for posterity: Use docker image: nagadomi/torch7

https://hub.docker.com/r/nagadomi/torch7