Skip to content

Instantly share code, notes, and snippets.

@typpo

typpo/neural-style-ec2.txt

Forked from elad/neural-style-ec2.txt
Created June 4, 2016 19:38
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save typpo/c5fe8346c401877b769fbca4cadea28f to your computer and use it in GitHub Desktop.
Save typpo/c5fe8346c401877b769fbca4cadea28f 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
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