A Dockerized, heavily modified version of the Volta Deep Style Transfer script made by Victor Espinoza (vic8760).
This is a Dockerized environment with CUDA 9.0, cuDNN v7, Torch7, Caffe, and all the required files pre-installed! Just launch with nvidia-docker and get dreaming!
sudo docker pull nimaid/volta-deep-style
sudo docker run --runtime=nvidia -it nimaid/volta-deep-style
The example script is volta-x1.sh. It's made for single-GPU machines, and uses only GPU 0 in multi-GPU environments.
./volta-x1.sh test_files/style/ test_files/content.jpg
Jupyter notebook tag and more code cleanup coming!
The Docker environment is completely functional, and you can simply load any existing scripts you have over to it and get running!
The Jupyter notebook is not really working very well yet, but you can get the idea of what I'm trying to do. For some reason, it's not giving the same results as with the volta-x1.sh. I really don't know much about multi-res transfers, just Docker. :) If you can lend a hand in helping figure out how to make my algorithm work better, please contact me!