【Linux】为Ubuntu配置CUDA+cuDNN环境

Nvidia 驱动安装CUDA 安装deb(local)安装方式deb(network)安装方式runfile(local)安装方式配置 CUDA 环境变量卸载 CUDAcuDNN安装Nvidia 驱动安装检查自己的电脑是否有 Nvidia 的独立显卡，可以在 NVIDA X Server Settings 中看到自己的显卡信息nvidia-settings以及英伟达系统管理接口（NVIDIA S

Henry-zhu

4118人浏览 · 2021-10-27 10:42:35

Henry-zhu · 2021-10-27 10:42:35 发布

Nvidia 驱动安装
CUDA 安装
cuDNN安装

Nvidia 驱动安装

检查自己的电脑是否有 Nvidia 的独立显卡，可以在 NVIDA X Server Settings 中看到自己的显卡信息

nvidia-settings

nvidia-settings

以及英伟达系统管理接口（NVIDIA System Management Interface, 简称 nvidia-smi）。这是是基于NVIDIA Management Library (NVML) 的命令行管理组件,旨在(intened to )帮助管理和监控NVIDIA GPU设备。
可以查看GPU使用情况，-l可以实时刷新

$ nvidia-smi 
Tue Jan  5 20:42:59 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.38       Driver Version: 455.38       CUDA Version: 11.1     |
|-------------------------------+----------------------+----------------------+
| 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  GeForce MX250       Off  | 00000000:01:00.0 Off |                  N/A |
| N/A   50C    P3    N/A /  N/A |    447MiB /  2002MiB |      3%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1109      G   /usr/lib/xorg/Xorg                 61MiB |
|    0   N/A  N/A      1656      G   /usr/lib/xorg/Xorg                190MiB |
|    0   N/A  N/A      1830      G   /usr/bin/gnome-shell               55MiB |
|    0   N/A  N/A      3382      G   ...AAAAAAAAA= --shared-files      101MiB |
|    0   N/A  N/A      3734      G   ...gAAAAAAAAA --shared-files       28MiB |
+-----------------------------------------------------------------------------+

我们可以看到我们显卡的型号 GeForce MX250，显存大小及使用情况 447MiB / 2002MiB，驱动版本 Driver Version: 455.38 及 CUDA 版本 CUDA Version: 11.1

查看 GPU 和推荐的驱动版本

ubuntu-drivers devices

然后进行自动安装

sudo ubuntu-drivers autoinstall

CUDA 安装

到NVIDIA官网下载CUDA toolkit，如果有需要可以参考阅读cuda toolkit release notes

Ubuntu下有三种安装方式

deb(local)安装方式
deb(network)安装方式
runfile(local)安装方式

如果需要安装TensorRT，貌似必须使用deb(local)方式安装，这种安装方式非常简单，只需要按照命令即可，并且强烈推荐这种安装方式

deb(local)安装方式

选择好需要安装的CUDA toolkit版本，再选择好对应系统、架构、发行版本及其版本、安装方式，安装对应命令即可

cuda-install-deb

例如，x86_64架构下的Ubuntu 20.04通过deb(local)的方式安装CUDA 11.4的命令如下

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
wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-11-4-local_11.4.0-470.42.01-1_amd64.deb
sudo apt-key add /var/cuda-repo-ubuntu2004-11-4-local/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda

deb(network)安装方式

选择好需要安装的CUDA toolkit版本，再选择好对应系统、架构、发行版本及其版本、安装方式，安装对应命令即可

例如，x86_64架构下的Ubuntu 20.04通过deb(network)的方式安装CUDA 11.4的命令如下

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

runfile(local)安装方式

选择好需要安装的CUDA toolkit版本，再选择好对应系统、架构、发行版本及其版本、安装方式，安装对应命令即可
例如，x86_64架构下的Ubuntu 20.04通过runfile(network)的方式安装CUDA 11.4的命令如下

wget https://developer.download.nvidia.com/compute/cuda/11.4.0/local_installers/cuda_11.4.0_470.42.01_linux.run
sudo sh cuda_11.4.0_470.42.01_linux.run

下面的配图是安装CUDA 11.0的版本
修改执行权限

sudo chmod 777 cuda_<cuda_version>_<driver_version>_linux.run

cuda install

执行安装程序

sudo ./cuda_<cuda_version>_<driver_version>_linux.run

cuda install

选择 Continue

cuda install

输入 accept 确认安装

cuda install

这里不要勾选 Drive ，因为这里安装的驱动可能低于系统自带的驱动，根据驱动安装

cuda install

安装结束后，出现安装概要

cuda install

===========
= Summary =
===========

Driver:   Not Selected
Toolkit:  Installed in /usr/local/cuda-11.0/
Samples:  Installed in /home/henryzhu/, but missing recommended libraries

Please make sure that
 -   PATH includes /usr/local/cuda-11.0/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-11.0/lib64, or, add /usr/local/cuda-11.0/lib64 to /etc/ld.so.conf and run ldconfig as root

To uninstall the CUDA Toolkit, run cuda-uninstaller in /usr/local/cuda-11.0/bin

Please see CUDA_Installation_Guide_Linux.pdf in /usr/local/cuda-11.0/doc/pdf for detailed information on setting up CUDA.
***WARNING: Incomplete installation! This installation did not install the CUDA Driver. A driver of version at least .00 is required for CUDA 11.0 functionality to work.
To install the driver using this installer, run the following command, replacing <CudaInstaller> with the name of this run file:
    sudo <CudaInstaller>.run --silent --driver

Logfile is /var/log/cuda-installer.log

这里的安装版本是CUDA 11.0，请根据自己的CUDA版本进行处理

配置 CUDA 环境变量

CUDA环境要求我们

Please make sure that
 -   PATH includes /usr/local/cuda-11.0/bin
 -   LD_LIBRARY_PATH includes /usr/local/cuda-11.0/lib64, or, add /usr/local/cuda-11.0/lib64 to /etc/ld.so.conf and run ldconfig as root

编辑环境变量文件

vim ~/.bashrc

添加以下内容

# ------ CUDA 11.0 ------
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export PATH=$PATH:/usr/local/cuda-11.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-11.0

这里以 CUDA 11.0 为例，请根据自己安装的CUDA版本进行对应修改cuda-11.0 修改为 cuda-<version>

执行命令使其生效

source ~/.bashrc

验证是否安装成功

$ nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Thu_Jun_11_22:26:38_PDT_2020
Cuda compilation tools, release 11.0, V11.0.194
Build cuda_11.0_bu.TC445_37.28540450_0

卸载 CUDA

sudo /usr/local/cuda-11.0/bin/cuda-uninstaller
sudo rm -rf /usr/local/cuda-11.0

cuDNN安装

下载 cuDNN（需要注册NVIDIA账号并登录）

cuDNN-download

下载解压之后

$ tree
.
└── cuda
    ├── include
    │   ├── cudnn_adv_infer.h
    │   ├── cudnn_adv_train.h
    │   ├── cudnn_backend.h
    │   ├── cudnn_cnn_infer.h
    │   ├── cudnn_cnn_train.h
    │   ├── cudnn.h
    │   ├── cudnn_ops_infer.h
    │   ├── cudnn_ops_train.h
    │   └── cudnn_version.h
    ├── lib64
    │   ├── ...
    │   ├── libcudnn_ops_train.so.8.0.4
    │   ├── libcudnn.so -> libcudnn.so.8
    │   ├── libcudnn.so.8 -> libcudnn.so.8.0.4
    │   ├── libcudnn.so.8.0.4
    │   └── libcudnn_static.a
    └── NVIDIA_SLA_cuDNN_Support.txt

3 directories, 32 files

将解压目录中的 cuda/include/cudnn.h 文件复制到 /usr/local/cuda-11.0/include 文件夹， cuda/lib64/ 下所有文件复制到 /usr/local/cuda-11.0/lib64 文件夹中

sudo cp cuda/include/* /usr/local/cuda-11.0/include
sudo cp cuda/lib64/* /usr/local/cuda-11.0/lib64

并添加读取权限

sudo chmod a+r /usr/local/cuda-11.0/include/cudnn*
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*

Linux

更多推荐

网卡速率和双工模式的配置

http://linux.chinaitlab.com/system/792187.html1、mii-tool 配置网络设备协商方式的工具； 1.1 mii-tool 介绍； mii-tool - view, manipulate media-independent interface status （mii-tool 是查看，管理介质的网络接口的状态）

Linux

Linux虚拟文件系统之文件系统卸载（sys_umount())

Linux中卸载文件系统由umount系统调用实现，入口函数为sys_umount()。较于文件系统的安装较为简单，下面是具体的实现。1. /*sys_umont系统调用*/2. SYSCALL_DEFINE2(umount, char __user *, name, int, flags)3. {4.struct path path;

Linux

Linux系统下超级终端Minicom的使用方法（例如：连接交换机，路由器等）转http://baike.baidu.com/view/2911642.htm?fr=ala0_1

Linux系统下超级终端Minicom的使用方法 　　Linux下的Minicom的功能与下的超级终端功能相似，适于在通过超级终端对设备的管理以及对嵌入操作系统的升级，现写出Minicom的使用手册： 　　1．启动minicom 　　以root权限登录系统 　　使用命令 　　minicom –s 则minicom启动，屏

Linux

所有评论(0)

查看更多评论

Henry-zhu

@weixin_44148028

已为社区贡献1条内容