*also works for Ubuntu 18.04
I?ll start my story with saying that I had issues like three years ago with CUDA 8.0 and Ubuntu 16.04. This time I?ll be smarter and document the process for my future self.
- Start terminal and remove any NVIDIA traces you may have on your machine.
sudo rm /etc/apt/sources.list.d/cuda*sudo apt remove –autoremove nvidia-cuda-toolkitsudo apt remove –autoremove nvidia-*
2. Setup the correct CUDA PPA on your system
sudo apt updatesudo add-apt-repository ppa:graphics-driverssudo apt-key adv –fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pubsudo bash -c ‘echo “deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64 /” > /etc/apt/sources.list.d/cuda.list’sudo bash -c ‘echo “deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64 /” > /etc/apt/sources.list.d/cuda_learn.list’
3. Install CUDA 10.1 packages
sudo apt updatesudo apt install cuda-10-1sudo apt install libcudnn7
4. As the last step one need to specify PATH to CUDA in ?.profile? file. Open the file by running:
sudo vi ~/.profile
And add the following lines at the end of the file:
# set PATH for cuda 10.1 installationif [ -d “/usr/local/cuda-10.1/bin/” ]; then export PATH=/usr/local/cuda-10.1/bin${PATH:+:${PATH}} export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}fi
5. Restart and check the versions for the installation.
CUDA:
nvcc ? versionnvcc: NVIDIA (R) Cuda compiler driverCopyright (c) 2005-2019 NVIDIA CorporationBuilt on Wed_Apr_24_19:10:27_PDT_2019Cuda compilation tools, release 10.1, V10.1.168
NVIDIA Driver:
nvidia-smi
libcudnn:
/sbin/ldconfig -N -v $(sed ?s/:/ /? <<< $LD_LIBRARY_PATH) 2>/dev/null | grep libcudnn
Enjoy CUDA coding!
Bonus track for everyone who wonders why it is so complicated:
