Skip to content

Instantly share code, notes, and snippets.

@olegakbarov
Forked from erikbern/install-tensorflow.sh
Created December 10, 2016 20:29
Show Gist options
  • Select an option

  • Save olegakbarov/4ff97c06065a913db8c5aa6d18af3ef1 to your computer and use it in GitHub Desktop.

Select an option

Save olegakbarov/4ff97c06065a913db8c5aa6d18af3ef1 to your computer and use it in GitHub Desktop.
Installing TensorFlow on EC2
sudo apt-get update
sudo apt-get upgrade -y # choose “choose package maintainers version”
sudo apt-get install -y build-essential linux-source linux-headers-`uname -r` python-pip python-dev
echo -e "blacklist nouveau\nblacklist lbm-nouveau\noptions nouveau modeset=0\nalias nouveau off\nalias lbm-nouveau off\n" | sudo tee /etc/modprobe.d/blacklist-nouveau.conf
echo options nouveau modeset=0 | sudo tee -a /etc/modprobe.d/nouveau-kms.conf
sudo update-initramfs -u
sudo reboot
wget http://developer.download.nvidia.com/compute/cuda/7.5/Prod/local_installers/cuda_7.5.18_linux.run
chmod +x cuda_7.5.18_linux.run
./cuda_7.5.18_linux.run -extract=`pwd`/nvidia_installers
cd nvidia_installers
sudo ./NVIDIA-Linux-x86_64-352.39.run
sudo modprobe nvidia
sudo ./cuda-linux64-rel-7.5.18-19867135.run
sudo pip install https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow-0.5.0-cp27-none-linux_x86_64.whl
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment