博客

在anaconda环境中配置cuda和cudnn

  • Post author:
  • Post category:其他

解决经常出现的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/

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

版权声明：本文为weixin_42727538原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。