Skip to content

Instantly share code, notes, and snippets.

@post2web
Forked from mjdietzx/waya-dl-setup.sh
Last active March 30, 2018 18:02
Show Gist options
  • Save post2web/4a5f556f36a3d2b61d603304f3838d25 to your computer and use it in GitHub Desktop.
Save post2web/4a5f556f36a3d2b61d603304f3838d25 to your computer and use it in GitHub Desktop.
Install CUDA 9, cuDNN v7.1, tensorflow-gpu on Ubuntu 16.04
# cuda 9
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64-deb
sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64-deb
sudo apt-key adv --fetch-keys \
http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-9-0
# cuDNN 7.05
# Get from cuDNN package (has login).
tar -xzvf cudnn-9.0-linux-x64-v7.tgz
# Copy the following files into the CUDA Toolkit directory.
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h
/usr/local/cuda/lib64/libcudnn*
# install pip
wget https://bootstrap.pypa.io/get-pip.py
sudo python3 get-pip.py
# tensorflow
sudo pip3 install tensorflow-gpu
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment