在anaconda环境中配置cuda和cudnn

解决经常出现的cuda版本不对应问题

首先查看自己设备允许安装的cuda版本

可使用 nvidia-smi 命令查看
在anaconda环境中配置cuda和cudnn_第1张图片
然后找到对应版本,注意这里的对应版本是指当前设备能安装的最高版本,可向下兼容。
在anaconda环境中配置cuda和cudnn_第2张图片

我们使用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/

这里的版本可以自己更改。

你可能感兴趣的:(深度学习,机器学习,python,linux)