Created
July 15, 2015 12:59
-
-
Save cr3ative/35240a472d6a95410c35 to your computer and use it in GitHub Desktop.
Vagrantfile for image-dreamer, more CPU, more RAM
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# -*- mode: ruby -*- | |
# vi: set ft=ruby : | |
Vagrant.configure(2) do |config| | |
config.vm.box = "data-science-toolbox/dst" | |
# Salt to taste | |
config.vm.provider "virtualbox" do |v| | |
v.memory = 12015 | |
v.cpus = 8 | |
end | |
config.vm.provision "shell", inline: <<-SHELL | |
sudo apt-get update | |
sudo apt-get install -y git bc wget | |
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libboost-all-dev libhdf5-serial-dev | |
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev protobuf-compiler | |
sudo apt-get install -y libopenblas-base libopenblas-dev | |
git clone https://github.com/BVLC/caffe.git | |
cd caffe | |
cp Makefile.config.example Makefile.config | |
sed -i "s/# CPU_ONLY := 1/CPU_ONLY := 1/" Makefile.config | |
sed -i "s/BLAS := atlas/BLAS := open/" Makefile.config | |
make -j`nproc` all | |
# Uncomment the below if you're feeling paranoid. We're not running in production or anything, here. | |
# make -j`nproc` test | |
# make -j`nproc` runtest | |
make -j`nproc` pycaffe | |
for req in $(cat python/requirements.txt); do sudo pip install $req; done | |
echo 'export PYTHONPATH=/home/vagrant/caffe/python:$PYTHONPATH' >> /home/vagrant/.profile | |
if [ ! -f "/home/vagrant/caffe/models/bvlc_googlenet/bvlc_googlenet.caffemodel" ]; then | |
wget -q http://dl.caffe.berkeleyvision.org/bvlc_googlenet.caffemodel -O /home/vagrant/caffe/models/bvlc_googlenet/bvlc_googlenet.caffemodel | |
fi | |
SHELL | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment