基于Ubuntu20.04 GTX960m搭建cuda&cunn

基础:

• PC:ubuntu 20.04
• 显卡:GTX 960m

获取电脑的显卡驱动型号

CUNN地址
CUDA-ToolKit
CUDA-ToolKit安装方式

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/7fa2af80.pub
sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
sudo apt-get update
sudo apt-get -y install cuda

通过nvcc -v获取PC是否已经安装成功CUDA

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Aug_15_21:14:11_PDT_2021
Cuda compilation tools, release 11.4, V11.4.120
Build cuda_11.4.r11.4/compiler.30300941_0

确认电脑已经安装成功ToolKit
通过nvidia-smi获取PC是否安装成功显卡驱动

Sat Sep 11 16:57:40 2021
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 470.63.01    Driver Version: 470.63.01    CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   37C    P8    N/A /  N/A |     64MiB /  4046MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A       984      G   /usr/lib/xorg/Xorg                 56MiB |
|    0   N/A  N/A      1383      G   /usr/bin/gnome-shell                6MiB |
+-----------------------------------------------------------------------------+

备注:ubuntu20.04可以通过软件和更新->附加驱动安装对应的官方驱动,这样简单些
备注:如果失败,请查看下是不是PC上的安全模式没有别关闭

然后根据需求的nvidia-smi获取的信息CUDA Version了解到对应的CUDA版本
安装成功后,下载对应的CUNN版本,对应的下载页面上对应版本的信息,下载

CUNN的配置

• 解压缩 cudnn-11.4-linux-x64-v8.2.2.26.tgz

tar -xzvf cudnn-11.4-linux-x64-v8.2.2.26.tgz

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

以上:仅仅是对于已有经历做了一个初步的记录,实际操作过程中,走了很多错误步骤,其中是否对于结果产生影响,个人能力有限,无法对比验证