Skip to content

Instantly share code, notes, and snippets.

@abdel
Forked from mjdietzx/waya-dl-setup.sh
Last active November 23, 2018 08:14
Show Gist options
  • Save abdel/2babdcc48ffc7fcdacc0980377353aa4 to your computer and use it in GitHub Desktop.
Save abdel/2babdcc48ffc7fcdacc0980377353aa4 to your computer and use it in GitHub Desktop.
Install CUDA Toolkit v9.0 and cuDNN v7.1 on Ubuntu 16.04
#!/bin/bash
# Install CUDA Toolkit v9.0
# Instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb (network))
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
CUDA_REPO_PKG="cuda-repo-ubuntu1604_9.0.176-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-9-0
# Install cuDNN v7.1
# Install instructions from https://developer.nvidia.com/cudnn (Membership Required)
# Download cuDNN v7.1.4 (May 16, 2018), for CUDA 9.0 -> cuDNN v7.1.4 Library for Linux
CUDNN_TAR_FILE="cudnn-9.0-linux-x64-v7.1.tgz"
tar -xzvf ${CUDNN_TAR_FILE}
sudo cp -P cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
sudo chmod a+r /usr/local/cuda-9.0/lib64/libcudnn*
# set environment variables
export PATH=/usr/local/cuda-9.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment