krystophv
9/7/2015 - 8:09 AM

Running neural-style in EC2

Running neural-style in EC2

Start a g2.2xlarge or better (GPU instance) with https://console.aws.amazon.com/ec2/v2/home?region=us-east-1#LaunchInstanceWizard:ami=ami-ffba7b94

Login, username is ubuntu

Update a bunch of stuff and make sure cudnn R2 is used:

luarocks install image
luarocks install loadcaffe
luarocks install torch
export LD_LIBRARY_PATH=/home/ubuntu/torch-distro/install/lib:/home/ubuntu/torch-distro/install/lib:/home/ubuntu/cudnn-6.5-linux-x64-v2-rc2

Get neural-style:

git clone https://github.com/jcjohnson/neural-style
cd neural-style
sh models/download_models.sh

Basic usage, note backend is cudnn:

th neural_style.lua -num_iterations 2000 -style_image style.jpg -content_image image.jpg -image_size 400 -backend cudnn -output_image output.png