Skip to content

Instantly share code, notes, and snippets.

@wasi

wasi/neural-style-ec2.txt

Forked from elad/neural-style-ec2.txt
Last active September 23, 2016 18:07
Show Gist options
  • Save wasi/bf626abd9ec6b152e1f49648652f2b4c to your computer and use it in GitHub Desktop.
Save wasi/bf626abd9ec6b152e1f49648652f2b4c to your computer and use it in GitHub Desktop.
Download ZIP
Running neural-style in EC2
Raw
neural-style-ec2.txt
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
luarocks install sys
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
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment