Skip to content

Instantly share code, notes, and snippets.

@mjdietzx
Last active March 13, 2024 15:08
Show Gist options
  • Save mjdietzx/0ff77af5ae60622ce6ed8c4d9b419f45 to your computer and use it in GitHub Desktop.
Save mjdietzx/0ff77af5ae60622ce6ed8c4d9b419f45 to your computer and use it in GitHub Desktop.
Install CUDA Toolkit v8.0 and cuDNN v6.0 on Ubuntu 16.04
#!/bin/bash
# install CUDA Toolkit v8.0
# instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb (network))
CUDA_REPO_PKG="cuda-repo-ubuntu1604_8.0.61-1_amd64.deb"
wget http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/${CUDA_REPO_PKG}
sudo dpkg -i ${CUDA_REPO_PKG}
sudo apt-get update
sudo apt-get -y install cuda
# install cuDNN v6.0
CUDNN_TAR_FILE="cudnn-8.0-linux-x64-v6.0.tgz"
wget http://developer.download.nvidia.com/compute/redist/cudnn/v6.0/${CUDNN_TAR_FILE}
tar -xzvf ${CUDNN_TAR_FILE}
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-8.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/
sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*
# set environment variables
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64\${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
@srhrshr
Copy link

srhrshr commented Jan 11, 2018

In case running nvidia-smi throws the following error, just try rebooting. That seemed to work for me.
NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running

This script works by the way! Ubuntu 16.04.3 represent here. I successfully installed Cuda 8.0 and cuDNN 6.0, without running into any problems with the display/graphics like I usually used to. I highly recommend using the package manager installation over the runfile method.

Thanks!

@wyfSunflower
Copy link

This script is not suitable for default graphic card is not nvidia card, such as " dell precision tower 7910", after I run this script and reboot ubuntu, system could not login. And now tensorflow can only support cuda8, but this script will install cuda9.1

@pbertsch
Copy link

For Ubuntu 16.04 users only:

change the the following line in the script
sudo apt-get -y install cuda to sudo apt-get -y install cuda-8-0

@JohnAllen
Copy link

GPU TensorFlow now requires cudnn 9.0. Anyone have an updated link for 9.0?

@anujonthemove
Copy link

If you are still facing installation issues, try out this guide: https://medium.com/@anujonthemove/deep-learning-environment-setup-on-ubuntu-16-04-83078e1cba1f

@prameshbajra
Copy link

GPU TensorFlow now requires cudnn 9.0. Anyone have an updated link for 9.0?

@JohnAllen Did you find a updated script for GPU tensorflow?

@jrichardsz
Copy link

4040 error on ubuntu 20

--2022-01-23 17:43:34--  http://developer.download.nvidia.com/compute/redist/cudnn/v6.0/cudnn-8.1-linux-x64-v6.0.tgz
Resolving developer.download.nvidia.com (developer.download.nvidia.com)... 152.199.20.126
Connecting to developer.download.nvidia.com (developer.download.nvidia.com)|152.199.20.126|:80... connected.
HTTP request sent, awaiting response... 301 Moved Permanently
Location: https://developer.download.nvidia.com/compute/redist/cudnn/v6.0/cudnn-8.1-linux-x64-v6.0.tgz [following]
--2022-01-23 17:43:34--  https://developer.download.nvidia.com/compute/redist/cudnn/v6.0/cudnn-8.1-linux-x64-v6.0.tgz
Connecting to developer.download.nvidia.com (developer.download.nvidia.com)|152.199.20.126|:443... connected.
HTTP request sent, awaiting response... 404 Not Found
2022-01-23 17:43:35 ERROR 404: Not Found.

tar (child): cudnn-8.1-linux-x64-v6.0.tgz: Cannot open: No such file or directory
tar (child): Error is not recoverable: exiting now
tar: Child returned status 2
tar: Error is not recoverable: exiting now
cp: cannot stat 'cuda/include/cudnn.h': No such file or directory
cp: cannot stat 'cuda/lib64/libcudnn*': No such file or directory
chmod: cannot access '/usr/local/cuda-8.0/lib64/libcudnn*': No such file or directory

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment