ubuntu 22.04配置cuda和cudnn

m0_521118232025-04-12 4:12

cuda：12.1

bash 复制代码

wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
sudo sh cuda_12.1.1_530.30.02_linux.run

Abort/Continue选择Continue，不要勾选自带的driver

配置环境变量。~/.bashrc

复制代码

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

安装补充包，动态加载和卸载 NVIDIA 内核模块

bash 复制代码

sudo apt install nvidia-modprobe

cudnn

bash 复制代码

sudo dpkg -i cudnn-local-repo-ubuntu2204-8.9.7.29_1.0-1_amd64.deb # 安装
sudo cp /var/cudnn-local-repo-ubuntu2204-8.9.7.29/cudnn-local-*-keyring.gpg /usr/share/keyrings/ # 导入GPG密钥
sudo apt update # 更新软件包列表
sudo apt install libcudnn8 libcudnn8-dev # 安装cuDNN运行时和开发库
sudo ldconfig -v # 刷新ldconfig 缓存
cat /usr/include/cudnn_version.h | grep CUDNN_MAJOR -A 2 # 检查cuDNN版本
ls /usr/lib/x86_64-linux-gnu/libcudnn*	# 检查库文件
sudo ldconfig -p | grep cudnn	# 测试动态链接

本文是转载文章，点击查看原文

如有侵权，请联系 xyy@jishuzhan.net 删除