【自用】在conda下安装不同的cuda版本（本次为cuda11.7）

言辰寸心2024-08-06 13:52

参考博客：

#查看当前cuda情况

bash 复制代码

nvcc -V

#查看cudatoolkit的可安装版本

bash 复制代码

conda search cudatoolkit --info

#安装cudatoolkit

bash 复制代码

conda install  cudatoolkit

#安装cuda11.7和对应的pytorch等环境

bash 复制代码

conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.7 -c pytorch -c nvidia

#查看包里面是否包含cuda 11.7

bash 复制代码

conda list

#需要在cudatoolkit安装之后！！！

bash 复制代码

conda install cudnn

#在python中查看cuda版本是否正确

bash 复制代码

python
>>>import torch
>>>print(torch.version.cuda) #11.7
>>>print(torch.backends.cudnn.version()) #8500

#运行时发现nvcc -V版本还是不合适(尝试在conda下安装nvcc)

#查找对应版本nvcc

bash 复制代码

 conda search -c nvidia cuda-nvcc

#安装nvcc

bash 复制代码

conda install -c nvidia cuda-nvcc==12.2.140 #或者是11.7.99