Ubuntu22.04下载显卡驱动和CUDA+cuDNN

随笔2周前发布 木屋
7 0 0

注:笔记本电脑-显卡 RTX4060 – CUDA11.8 – cuDNN8.9.7

 

1:显卡驱动

已经安装驱动的,以此卸载已下载驱动

sudo apt-get remove  --purge nvidia-*

 新机器直接进行以下

# 查看驱动列表
sudo apt install aptitude sudo aptitude search nvidia-driver-*

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

显卡驱动一般选择最新版,后缀server服务器版本,可进行多卡集成作业 

# 回车后继续 
sudo apt-get update 
sudo apt-get install nvidia-dirver-版本号-server

之后重启系统让显卡驱动生效
终端输入 : nvidia-smi

 

 

2:CUDA版本的选择与安装

CUDA各版本下载网址
https://developer.nvidia.com/cuda-toolkit-archive/ 
cuDNN各版本下载网址
https://developer.nvidia.com/rdp/cudnn-archive#a-collapse805-111
或者
https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/

我选择CUDA11.8+cuDNN8.9.7.29

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

 

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

选择continue

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

 输入accept

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

 按空格取消选择第一项(Nvidia驱动已安装)

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

最后install

 

配置环境变量

# 配置环境变量:在家目录下 ~/.bashrc 的最下面添加:
# CUDA库函数的路径
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
# CUDA头文件的路径
export CPLUS_INCLUDE_PATH=/usr/local/cuda-11.8/include:$CPLUS_INCLUDE_PATH
# CUDA家目录
export CUDA_HOME=/usr/local/cuda-11.8
# CUDA可执行命令的路径
export PATH=/usr/local/cuda-11.8/bin:$PATH
# 使环境生效
source ~/.bashrc

 

 3:cuDNN安装

安装完成后可用 nvcc -V 命令验证是否安装成功

tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

# 进入cuda版本目录,建立软链接
cd /usr/local/cuda-11.8
sudo ln -sf libcudnn.so.8.9.7 libcudnn.so.8
sudo ln -sf libcudnn.so.8 libcudnn.so
sudo tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz -C /usr/local/ -v

 

# 方法不是所有人都适用,若有问题,或可尝试以下
#安装cudnn是通过tar包进行安装的,注意有软链接的问题。不能只cp过去,软链接复制过去就变了。
tar -xvf cudnn-linux-x86_64-8.9.7.29_cuda11-archive.tar.xz
$ cd cudnn* (cudnn减压后的目录)
$ sudo cp include/* /usr/local/cuda/include
$ cd lib
$ tar czvf ../lib64.tar *  (把lib里的所有文件进行压缩,放到上一层目录下,名字叫lib64.tar)
$ sudo tar -xvf ../lib64.tar -C /usr/local/cuda/lib64 -v  (保证lib里边的软连接没有变)

 

验证

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

 

Ubuntu22.04下载显卡驱动和CUDA+cuDNN

 

© 版权声明

相关文章

暂无评论

您必须登录才能参与评论!
立即登录
暂无评论...