解决经常出现的cuda版本不对应问题
首先查看自己设备允许安装的cuda版本
可使用 nvidia-smi 命令查看
然后找到对应版本,注意这里的对应版本是指当前设备能安装的最高版本,可向下兼容。
我们使用conda命令安装cuda和cudnn
conda install cudatoolkit=10.0 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
conda install cudnn=7.3.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
这里的版本可以自己更改。