当前位置:网站首页>在anaconda环境中配置cuda和cudnn

在anaconda环境中配置cuda和cudnn

2022-08-09 09:28:00 Huranqingqing

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

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

原网站

版权声明
本文为[Huranqingqing]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_42727538/article/details/126194941