-
cuda:12.1
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 内核模块
sudo apt install nvidia-modprobe
-
cudnn
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 # 测试动态链接