dpit/neural-style-gpu repository overview

Please install NVIDIA Docker first: https://github.com/NVIDIA/nvidia-docker⁠.

To run it: nvidia-docker run --rm -it dpit/neural-style-gpu

To test that it works: cd ~/neural-style th neural_style.lua -gpu 0 -print_iter 1 You should see the output as described at https://github.com/jcjohnson/neural-style/blob/master/INSTALL.md⁠. Or run neural-style with cuDNN like this: th neural_style.lua -gpu 0 -backend cudnn

For further details see https://github.com/jcjohnson/neural-style⁠.