Ubuntu 22.04LTS + 深度学习环境安装全流程 cuda12.2

步骤1. 安装cuda toolkit

选择驱动、toolkit对应的版本 https://developer.nvidia.com/cuda-toolkit-archive

wget https://developer.download.nvidia.com/compute/cuda/12.2.0/local_installers/cuda_12.2.0_535.54.03_linux.run
sudo sh cuda_12.2.0_535.54.03_linux.run

提醒 ==> 配置环境变量

vim ~/.bashrc

# 添加一行
export PATH=/usr/local/cuda-12.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

# 保存并加载环境变量
source ~/.bashrc

 

步骤2. 安装驱动

CUDA Toolkit和 显卡驱动 的版本对应

注意在安装Tool kit过程中可以选择安装驱动，但过程复杂，选择自动安装显卡驱动（安装完需重启机器）

sudo ubuntu-drivers autoinstall

或者直接输入nvidia-smi会给出安装指令及对应版本 ==> apt install *-server

 

步骤3. 安装cuDNN

从链接 cuDNN Download下载与cuda对应的cuDNN

将下载文件上传到ubuntu路径

scp /path/.../cudnn-local-repo-ubuntu2204-8.9.5.30_1.0-1_amd64.deb host_name@host_op:/home/path

 安装cuDNN

apt install ./cudnn-local-repo-ubuntu2204-8.9.5.30_1.0-1_amd64.deb.deb

 

步骤4. 安装pytorch

参考链接 pytorch与cuda对应关系

conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 pytorch-cuda=12.1 -c pytorch -c nvidia

 

 

 

参考资料

1. NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver
2. Ubuntu 22.04LTS + 深度学习环境安装全流程
3. Ubuntu22.04搭建Pytorch框架深度学习环境+安装Miniconda+安装CUDA与cuDNN
4. pip config set global.index error
5. Ubuntu22.04配置深度学习环境
6. 如何在linux中卸载CUDA
7. linux卸载cuda、cudnn
8. 图文并茂讲解CUDA, CUDA Toolkit, CUDA Driver, CUDA Runtime, nvcc之间的关系及其版本兼容性
9. 显卡驱动下载地址