conda虚拟环境中安装cuda和cudnn，再也不用头疼版本号的问题了

conda search cudatoolkit --info

执行上述命令后，会显示出源内所有的cuda版本，以及下载地址，如图所示

下载cuda

找到自己想要的cuda版本后，先把cuda下载到本地：

首先复制url字段里的下载链接，然后用命令cd到想要下载的目录，执行如下代码下载

wget 你刚刚复制的链接地址

安装cuda

执行如下命令进行安装，因为是通过本地安装的，所以需要写明本地包的路径

# 然后安装本地包
conda install --use-local 本地cuda包所在的路径/包名

#比如

conda install --use-local ./cudatoolkit-11.8.0-hd77b12b_0.conda

 安装pytorch

首先安装与cuda相匹配的pytorch和torchvision包，可以去pytorch官网查询。

因为我的cuda版本是11.8，支持的pytorch版本为2.0.1，所以我安装的是这俩个，你可以根据你的需要安装。

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.8 -c pytorch -c nvidia

可以用清华源加速

conda源国内只有清华有，
修改源只需输入如下两行命令：


conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes 

测试cuda版本

装好pytorch后，命令行输入python，进入python的命令行，导入torch包

查询cuda版本

print(torch.version.cuda)

得到如图结果，成功

conda虚拟环境中安装cuda和cudnn，再也不用头疼版本号的问题了_conda cudnn-CSDN博客