Public Repository

Last pushed: 4 months ago
Short Description
Torch implementation of neural style algorithm.
Full Description

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.

Docker Pull Command
Owner
dpit

Comments (0)