Detailed tutorial on installing nvidia driver + CUDA + cuDNN in Ubuntu 16.04

Preparation

1. Check whether the GPU supports CUDA

lspci | grep -i nvidia

2. Check the Linux version

uname -m && cat /etc/*release

nvidia drivers

1. Uninstall the original N card driver first

#for case1: original driver installed by apt-get:
sudo apt-get remove --purge nvidia*
#for case2: original driver installed by runfile:
sudo chmod +x *.run
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

2. Disable nouveau driver

sudo gedit /etc/modprobe.d/blacklist.conf

Add at the end of the text:

blacklist nouveau
options nouveau modeset=0

Then execute:

sudo update-initramfs -u

After restarting, you can check whether nouveau is running:

lsmod | grep nouveau # 沒輸出代表禁用生效

3. Disable X-Window Service

sudo service lightdm stop #This will shut down the graphical interface

Press Ctrl-Alt+F1 to enter the command line interface and enter your user name and password to log in.

4. Install the driver via command line

Driver URL: https://www.nvidia.cn/Download/index.aspx?lang=cn

#Give execution permission to the driver run file:
sudo chmod +x NVIDIA-Linux-x86_64-384.59.run
#The following parameters are very important and cannot be omitted:
sudo ./NVIDIA-Linux-x86_64-384.59.run –no-opengl-files

5.Driver test:

nvidia-smi #If the GPU information list is listed, it means the driver is installed successfully
nvidia-settings #If the settings dialog box pops up, it also means that the driver has been successfully installed.

CUDA Installation

1. Download CUDA for the corresponding system

URL: http://developer.nvidia.com/cuda-downloads Select runfile to install

2. Installation

sudo sh cuda_<version>_linux.run

After starting the installation, you need to read the instructions. You can press Ctrl + C to finish reading directly, or use the space bar to read slowly. The following are the installation options:

(Do you agree to the terms? You must agree to continue the installation)
accept/decline/quit: accept
(Do not install the driver here, because the latest driver has been installed. Otherwise, an old version of the graphics driver may be installed, resulting in repeated logins)
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 410.48?
(y)es/(n)o/(q)uit: n
Install the CUDA 10.0 Toolkit? (Do you want to install CUDA 10? This must be installed.)
(y)es/(n)o/(q)uit: y
Enter Toolkit Location (installation path, use the default, just press Enter)
[ default is /usr/local/cuda-10.0 ]: 
Do you want to install a symbolic link at /usr/local/cuda? (Agree to create a soft link)
(y)es/(n)o/(q)uit: y
Install the CUDA 10.0 Samples? (No need to install the test, it is already there)
(y)es/(n)o/(q)uit: n
Installing the CUDA Toolkit in /usr/local/cuda-10.0 ...

3. Configure environment variables

sudo gedit ~/.bashrc

Join at the end

export PATH=/usr/local/cuda-8.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

renew

source ~/.bashrc

4. Verify the installation

Check the cuda version

nvcc -V
CUDA Sample test:
#Compile and test deviceQuery:
cd /usr/local/cuda-8.0/samples/1_Utilities/deviceQuery
make
./deviceQuery
#Compile and test bandwidthTest:
cd ../bandwidthTest
make
./bandwidthTest

If the final results of these two tests are both Result = PASS, it means that CUDA is installed successfully.

*5. CUDA uninstall

Type in the command line

sudo apt-get remove cuda 
sudo apt-get autoclean
sudo apt-get remove cuda*

Then switch to the directory usr/local/

cd /usr/local/
sudo rm -r cuda-9.1

cuDNN Installation

Download the corresponding version of cuDNN https://developer.nvidia.com/cudnn

tar xvzf cudnn-9.2-linux-x64-v7.1
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig

Reference

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#post-installation-actions

Summarize

The above is the tutorial on how to install nvidia driver + CUDA + cuDNN in Ubuntu 16.04. I hope it will be helpful to you. If you have any questions, please leave me a message and I will reply to you in time. I would also like to thank everyone for their support of the 123WORDPRESS.COM website!
If you find this article helpful, please feel free to reprint it and please indicate the source. Thank you!