Skip to content

Instantly share code, notes, and snippets.

@elad
Created September 7, 2015 08:09
Show Gist options
  • Save elad/e3a9e3cc609996b13454 to your computer and use it in GitHub Desktop.
Save elad/e3a9e3cc609996b13454 to your computer and use it in GitHub Desktop.
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
@protivinsky
Copy link

Awesome, thanks! I only needed to add

luarocks install sys

otherwise I was not able to run th neural_style.lua.

@ProGamerGov
Copy link

ProGamerGov commented May 4, 2016

You need to put it in more regions for when individuals and/or companies are driving up the prices in US EAST.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment