[笔记] Ubuntu 18.04安装cuda 10及cudnn 7流程

2023-08-02 19:56:10

安装环境

OS：Ubuntu 18.04 64 bit
显卡：NVidia GTX 1080

任务：安装 CUDA 10及cuDNN 7

工具下载

NVidia官网下载下列文件：

CUDA 10：cuda_10.0.130_410.48_linux.run

cnDNN 7.4：cudnn-10.0-linux-x64-v7.4.2.24.tgz

安装CUDA

$ sudo sh cuda_10.0.130_410.48_linux.run

先输入accept接受协议，然后按需回答问题即可。

注意：

如果当前显卡驱动版本高于CUDA安装包内的驱动，建议跳过这一步，保留原来的显卡驱动即可
为了节省空间，sample可以不装

接着将下面内容追加到~/.bashrc：

export PATH=/usr/local/cuda-10.0/bin:$PATH

验证CUDA

使用nvidia-smi查看驱动版本为415.23：

$ nvidia-smi

Thu Jan 24 18:00:52 2019

+-----------------------------------------------------------------------------+

| NVIDIA-SMI 415.23       Driver Version: 415.23       CUDA Version: 10.0     |

|-------------------------------+----------------------+----------------------+

| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |

| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |

|===============================+======================+======================|

|   0  GeForce GTX 1080    Off  | 00000000:01:00.0  On |                  N/A |

| 48%   35C    P8     8W / 180W |    200MiB /  8116MiB |      2%      Default |

+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+

| Processes:                                                       GPU Memory |

|  GPU       PID   Type   Process name                             Usage      |

|=============================================================================|

|    0      1462      G   /usr/lib/xorg/Xorg                            83MiB |

|    0      1600      G   /usr/bin/gnome-shell                         100MiB |

|    0      1794      G   /opt/teamviewer/tv_bin/TeamViewer             14MiB |

+-----------------------------------------------------------------------------+

CUDA版本为10.0：

$ nvcc --version

nvcc: NVIDIA (R) Cuda compiler driver

Copyright (c) 2005-2018 NVIDIA Corporation

Built on Sat_Aug_25_21:08:01_CDT_2018

Cuda compilation tools, release 10.0, V10.0.130

安装cuDNN

流程是：解压，拷贝，配置环境变量

# 解压

$ tar -zxvf cudnn-10.0-linux-x64-v7.4.2.24.tgz

# 拷贝

$ cd cudnn-10.0-linux-x64-v7.4.2.24

$ sudo cp cuda/include/cudnn.h /usr/local/cuda-10.0/include

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda-10.0/lib64

# 修改权限

$ sudo chmod a+r /usr/local/cuda-10.0/include/cudnn.h /usr/local/cuda-10.0/lib64/libcudnn*

将下面内容追加到~/.bashrc：

export PATH=/usr/local/cuda-10.0/bin:$PATH

export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64:$LD_LIBRARY_PATH

export CUDNN_PATH="/usr/local/cuda-10.0/lib64/libcudnn.so"

验证cnDNN

下面命令不报错就OK。

$ echo -e '#include"cudnn.h"\n void main(){}' | nvcc -x c - -o /dev/null -lcudnn

码农公寓