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