titu1994 / Neural-Style-Transfer

Keras Implementation of Neural Style Transfer from the paper "A Neural Algorithm of Artistic Style" (http://arxiv.org/abs/1508.06576) in Keras 2.0+
Apache License 2.0
2.26k stars 481 forks source link

Easy Linux Install #54

Closed minihat closed 5 years ago

minihat commented 5 years ago

This project is, by far, the easiest to set up of the nn style transfer repos I've encountered. It might be nice to include in the guide a brief example of a conda environment setup that works on Ubuntu (with anaconda):

conda create -n nst numpy scipy pil h5py keras tensorflow-gpu

As long as you have an NVIDIA GPU and latest driver installed ahead of time, this will relieve you of the need to manually install CUDA and cuDNN.

titu1994 commented 5 years ago

Seems like a good idea. If you'd like credit, it would be great if you add a PR to update the readme or the guide.md.