Skip to content

Instantly share code, notes, and snippets.

@beci
Created June 1, 2017 19:42
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 beci/f34adcfcdbdbea7fad336e2e5196807c to your computer and use it in GitHub Desktop.
Save beci/f34adcfcdbdbea7fad336e2e5196807c to your computer and use it in GitHub Desktop.
tensorflow in aws
sudo apt-get update
sudo apt-get --assume-yes upgrade
sudo apt-get --assume-yes install tmux htop p7zip-full build-essential gcc g++ make binutils
sudo apt-get --assume-yes install software-properties-common
wget https://developer.nvidia.com/compute/cuda/8.0/Prod2/local_installers/cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
wget https://repo.continuum.io/archive/Anaconda3-4.4.0-Linux-x86_64.sh
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64-deb
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
#wget https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v6/prod/8.0_20170427/Ubuntu16_04_x64/libcudnn6_6.0.21-1+cuda8.0_amd64-deb
#use scp to copy instead
scp -i ec2_ohio2.pem Downloads/libcudnn6_6.0.21-1+cuda8.0_amd64.deb ubuntu@ec2-13-58-184-236.us-east-2.compute.amazonaws.com:/home/ubuntu
#install with apt or dpkg
CUDA_HOME
export CUDA_HOME=/usr/local/cuda-8.0${CUDA_HOME:+:${CUDA_HOME}}
conda create -n tf
pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.1.0-cp36-cp36m-linux_x86_64.whl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment