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:
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.