如何在conda虚拟环境中安装所需版本的cuda和cudnn

创建好虚拟环境后，激活环境，

（1）在环境中安装cuda

conda install cudatoolkit=你所需要的版本（例如 cudatoolkit=10.1）

 
 

 
 

若安装不成功则使用镜像源安装：

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

 
 

 
 

（2）在环境中安装cudnn

conda install cudnn=你所需要的版本（例如cudnn=7.6.5）

 
 

 
 

若安装不成功则使用镜像源安装：

conda install cudnn=7.6.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

 
 

 
 

如果在安装的时候没有写版本，则会自己匹配与cuda版本相匹配的版本。

如果你不想用conda环境，使用本地安装：
https://blog.csdn.net/qq_43705697/article/details/121618276