前言:
CUDA及cuDNN的安装(linux)Linux下安装显卡驱动及CUDA程序
系统参数:
系统:ubuntu20.04.2
显卡: GeForce GTX 1050
需要安装的版本:CUDA 11.4
一、 安装显卡驱动
(1) 检测你的NVIDIA图形卡和推荐的驱动程序的模型
ubuntu-drivers devices
(2) 选择安装所有推荐的驱动
sudo ubuntu-drivers autoinstall
(3) nvidia-smi 检查安装驱动是否成功
结果显示:
NVIDIA-SMI 470.74 Driver Version: 470.74 CUDA Version: 11.4
二、 安装CUDA程序
(1)下载cuda:
进入NVIDIA下载安装包 CUDA Toolkit 11.7 Downloads | NVIDIA Developer
选择相应的版本,有相关的命令 在终端输入即可 (注意:输入自己的安装版本)
wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
(2) cuda安装:
sudo sh cuda_11.0.2_450.51.05_linux.run
(3)cuda 环境变量配置:
.1. 安装vim包
sudo apt install vim # version 2:8.1.2269-1ubuntu5.4, or
sudo apt install vim-tiny # version 2:8.1.2269-1ubuntu5.4
sudo apt install neovim # version 0.4.3-3
sudo apt install vim-athena # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-gtk3 # version 2:8.1.2269-1ubuntu5.4
sudo apt install vim-nox # version 2:8.1.2269-1ubuntu5.4
.2. 设置环境变量 gedit ~/.bashrc
推荐用这个,打开是文件好操作
vim ~/.bashrc
打开文件,把以下的输入进去:(注意: i
插入 :
wq
退出
)
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=$PATH:/usr/local/cuda/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda
最后输入以下,更新环境变量
source ~/.bashrc
(4) 检查安装是否成功
nvcc -V
三、 安装cudnn程序
(1) cudnn下载
cuDNN是GPU加速计算深层神经网络的库。首先去官网 cuDNN Archive | NVIDIA Developer(2) 安装
解压缩->在压缩后的
cuda文件夹,打开终端
#注意,解压后的文件夹 将对应文件复制到 /usr/local中的cuda内
sudo cp cuda/include/cudnn.h /usr/local/cuda-11.0/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.0/lib64/
sudo chmod a+r /usr/local/cuda-11.0/include/cudnn.h
sudo chmod a+r /usr/local/cuda-11.0/lib64/libcudnn*