1.安装驱动
ubuntu-drivers devices
可以看到建议安装驱动为nvidia-driver-525版本
sudo apt install nvidia-driver-525
或者
sudo ubuntu-drivers autoinstall
输入nvidia-smi
查看驱动是否安装成功
2.安装cuda
2.1 下载cuda的runfile版本
在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
输入:
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
如果之前安装过驱动,这里的驱动选项可以不选。
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
查看驱动是否安装完成:
2.3 多个cuda版本切换的方法
此种安装方式,是将/usr/local的cuda文件链接到了对应cuda版本的根目录。输入以下命令可以查看当前cuda文件的链接情况:
cd /usr/local/
stat cuda
随后输出以下信息:
ls查看系统上安装的cuda版本(可以有多个,只需要修改链接指向即可):
命令行输入以下命令,断开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版本:
解压下载好的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安装完毕。
标签:local,ubuntu22.04,sudo,cudnn,cuda,安装,usr From: https://www.cnblogs.com/raiuny/p/16963325.html