conda虚拟环境中安装cuda、cudnn、pytorch

Ubuntu 22.04 LTS

安装NVIDIA显卡驱动后，查看自己的cuda版本

nvidia-smi

我们所说的安装cuda版本其实是安装cudatoolkit版本，这里的cuda版本是12.0，所以只要安装的cudatoolkit版本<=12.0即可

以安装cudatookit11.3.1、cudnn8.2.1、pytorch1.11.0为例

创建虚拟环境

conda create -n miku python=3.9

激活虚拟环境

conda activate miku

查看conda里有哪些cuda版本

conda search cudatoolkit --info

有我们需要的11.3.1版本，下载

conda install cudatoolkit=11.3.1

查看conda里有哪些cudnn版本

conda search cudnn --info

有我们需要的8.2.1版本，下载

conda install cudnn=8.2.1

 也可以用pip下载8.9.4版本（当前最新版）

pip install nvidia-cudnn-cu11==8.9.4.26

安装pytorch

pytorch官网Previous PyTorch Versions | PyTorch

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch

检查是否安装完成

python

import torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())

安装成功