Loading

ubuntu22.04 cuda和cudnn安装和配置

1.安装驱动

ubuntu-drivers devices

image

可以看到建议安装驱动为nvidia-driver-525版本

sudo apt install nvidia-driver-525

或者

sudo ubuntu-drivers autoinstall

输入nvidia-smi查看驱动是否安装成功


2.安装cuda

2.1 下载cuda的runfile版本

image

NVIDIA官网查看自己显卡对应的cuda版本信息。
我的下载链接在这:
https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=runfile_local
image

输入:

wget https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux.run

下载cuda的run版本,然后安装到/usr/local/cuda目录下:

sh XX_linux.run --silent --toolkit --samples --librarypath=/usr/local/cuda

image

如果之前安装过驱动,这里的驱动选项可以不选。

2.2 配置环境变量

vim ~/.bashrc
在后面添加:

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

source ~/.bashrc使之生效。
nvcc -V查看驱动是否安装完成:
image


2.3 多个cuda版本切换的方法

此种安装方式,是将/usr/local的cuda文件链接到了对应cuda版本的根目录。输入以下命令可以查看当前cuda文件的链接情况:
cd /usr/local/
stat cuda
随后输出以下信息:
image

ls查看系统上安装的cuda版本(可以有多个,只需要修改链接指向即可):
image

命令行输入以下命令,断开cuda文件与cuda-10.1的符号链接:
sudo rm -rf cuda
命令行输入以下命令,建立cuda文件与cuda-9.0的符号链接:
sudo ln -s /usr/local/cuda-9.0 /usr/local/cuda
此时nvcc -V查看cuda版本,已经更改成功,更换其他版本同样操作即可。


3.安装cudnn

官网下载cuda对应版本的cudnn:
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
这里我们选择的是最新的cudnn版本v8.6.0。
进入后选择tar版本:
image

解压下载好的tar文件。

tar -xvf xxxx.tar

将相应文件复制到对应cuda的目录:

cd cudnn-linux-x86_64-8.6.0.163_cuda11-archive
sudo cp include/cudnn.h    /usr/local/cuda/include
sudo cp lib/libcudnn*    /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h   /usr/local/cuda/lib64/libcudnn*

至此cudnn安装完毕。

posted @ 2022-12-08 13:46  raiuny  阅读(7575)  评论(0编辑  收藏  举报