Skip to content

Instantly share code, notes, and snippets.

@liepieshov
Last active January 4, 2022 03:38
Show Gist options
  • Save liepieshov/961005403e7c824addd225fc20377db2 to your computer and use it in GitHub Desktop.
Save liepieshov/961005403e7c824addd225fc20377db2 to your computer and use it in GitHub Desktop.
Installation guide cuda-10-0 with cudnn 7.5 on Ubuntu (for Tensorflow/Pytorch)

Installation guide cuda-10-0 with cudnn 7.5 on Ubuntu (for Tensorflow/Pytorch)

Remove previous versions of CUDA:

sudo apt-get purge nvidia*
sudo apt-get autoremove
sudo apt-get autoclean
sudo rm -rf /usr/local/cuda*

Install cuda

Go to Nvidia site here https://developer.nvidia.com/cuda-10.0-download-archive
Choose your:

  1. Platform
  2. Architecture
  3. Distribution
  4. Version
  5. In the last step choose dev (network)

Download the file

# for Ubuntu18.04
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb

Then

sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda-10-0

Reboot

sudo reboot

Add cuda to the $path

echo 'export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc
sudo ldconfig

Install cudNN

Go to the site https://developer.nvidia.com/rdp/cudnn-archive
Download cuDNN v7.5.0 (Feb 21, 2019), for CUDA 10.0, cuDNN Library for Linux. So you should have cudnn-10.0-linux-x64-v7.5.0.56.tgz
Then copy necessary files

tar -xf cudnn-10.0-linux-x64-v7.5.0.56.tgz
sudo cp -R cuda/include/* /usr/local/cuda-10.0/include
sudo cp -R cuda/lib64/* /usr/local/cuda-10.0/lib64
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment