我要评论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
检查是否安装完成
pythonimport torch
print(torch.version.cuda)
print(torch.backends.cudnn.version())
安装成功
发表评论