Logo

Indie Machine Learning and Video Game Dev

Neural Net Downloads
How to
Hardware Reviews
Get Merch
About
Training of the week

Installing CUDA 10 on Ubuntu 18.04/18.10

by Mathieu Poliquin

#Installing CUDA 10 on Ubuntu 18.04/18.10

Assuming a fresh install of 18.04 or 18.10

You can follow instructions for 18.04 on the tensorflow gpu install page but it also works for 18.10: CUDA Install

I copied the instructions I used here but added some additional trouble shooting notes

1. Add NVIDIA package repositories

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
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
wget http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1804/x86_64/nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb
sudo apt-get update

2. Install driver

In the software update you might not see the 418 driver available, if that is the case refresh the available drivers with the following commands

sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update

After that you can install it this way as stated in tensorflow installation guide

sudo apt-get install --no-install-recommends nvidia-driver-418

On some systems you might not be able to install 418 as it complains abut missing dependencies that also cannot be installed because they don’t support your system. The workaround is to install 430 version which compatible. But for that you might need to get the source package first:

sudo apt-get install nvidia-kernel-source-430
sudo apt-get install nvidia-kernel-common-430
sudo apt-get install nvidia-driver-430

As an alternative you can also use the Software and Update app if the command doesn’t work some reasons ![driver_list(/assets/ubuntu/driver_list.png)

Don’t forget to REBOOT!!!!!

Install development and runtime libraries

sudo apt-get install --no-install-recommends cuda-10-0 libcudnn7=7.6.0.64-1+cuda10.0 libcudnn7-dev=7.6.0.64-1+cuda10.0

4. Tensorflow

As of this writing the latest version of Tensorflow is 1.14 which supports CUDA 10

pip3 install tensorflow-gpu

You can specify a version by adding ==[version]

pip3 install tensorflow-gpu==1.14
tags: cuda - 10 - ubuntu - 18.04 - 18.1 - machine learning