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

如何在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版本相匹配的版本。