-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathVagrantfile
65 lines (52 loc) · 2.32 KB
/
Vagrantfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# -*- mode: ruby -*-
# vi: set ft=ruby :
VAGRANTFILE_API_VERSION = "2"
Vagrant.configure(VAGRANTFILE_API_VERSION) do |config|
config.vm.box = "dummy"
config.vm.box_url = "https://github.com/mitchellh/vagrant-aws/raw/master/dummy.box"
# work around for vagrant-aws v0.7.2 and vagrant v1.9.1.
# Please see https://github.com/mitchellh/vagrant/issues/8118
config.ssh.pty = false
config.vm.provider :aws do |aws, override|
aws.access_key_id = ENV['AWS_ACCESS_KEY_ID']
aws.secret_access_key = ENV['AWS_SECRET_ACCESS_KEY']
# If you want to use other region,
# please find an appropriate public ami from
# https://cloud-images.ubuntu.com/locator/ec2/
aws.region = "ap-northeast-1"
aws.ami = "ami-3995e55e"
aws.instance_type="g2.2xlarge"
aws.keypair_name = "KEYPAIR NAME"
aws.security_groups = "SECURITY GROUP NAME"
override.ssh.username = "ubuntu"
override.ssh.private_key_path = "PATH TO YOUR PRIVATE KEY"
end
config.vm.provision "shell", inline: <<SCRIPT
apt-get update && apt-get -y upgrade
apt-get -y install linux-headers-$(uname -r) linux-image-extra-$(uname -r)
# Install CUDA v8.0
dpkg --configure -a
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_8.0.44-1_amd64.deb
dpkg -i cuda-repo-ubuntu1404_8.0.44-1_amd64.deb
apt-get update && apt-get install -y cuda-8-0
rm cuda-repo-ubuntu1404_8.0.44-1_amd64.deb
# Install cuDNN v5.1
wget http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz
tar xvzf cudnn-8.0-linux-x64-v5.1.tgz*
rm cudnn-8.0-linux-x64-v5.1.tgz
cp cuda/include/cudnn.h /usr/local/cuda/include/
cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
rm -rf /home/ubuntu/cuda
# Setup library paths
echo 'export CUDA_HOME=/usr/local/cuda
export CUDA_ROOT=/usr/local/cuda
export PATH=\$PATH:\$CUDA_ROOT/bin:$HOME/bin
export LD_LIBRARY_PATH=\$LD_LIBRARY_PATH:\$CUDA_ROOT/lib64
' >> /home/ubuntu/.bashrc
# Install tensorflow for gpu
# Please see https://github.com/tensorflow/tensorflow/blob/master/tensorflow/g3doc/get_started/os_setup.md
curl https://bootstrap.pypa.io/get-pip.py | sudo python3
pip3 install --no-cache-dir https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-0.12.1-cp34-cp34m-linux_x86_64.whl
SCRIPT
end