作为一个计算机视觉的忠实粉丝,运行及处理速度一直是最关心也是最让人头疼的问题,今天就来说说我的win10+tensorflow-GPU+cuda10.0+cudnn7.41安装过程,也是一段辛酸史,因为这对一个小白来讲真的痛苦到绝望。。
首先说明一下需要下载的东西:
1、cuda(版本需要结合你的计算机的显卡型号及计算能力)
2、cudnn它的下载完全是为了对cuda的补充而且不可或缺,也是需要与cuda版本相对应
正题:
cuda下载及安装+变量配置
你需要查看自己的电脑显卡型号下载支持的cuda版本与对应cudnn版本
选中此电脑-右击属性
官网的显卡型号是否支持查询网址:https://developer.nvidia.com/
关于计算机显卡NVIDIA的型号与计算能力如下:
一般情况下计算能力大于3.1的都可以
然后选择适合自己要安装的tensorflow-gpu版本的CUDA和cuDNN,这个一般在tensorflow官网查看信息(建议计算能力不高的电脑不选择高版本):
下载安装CUDA
进入网址:https://developer.nvidia.com/cuda-toolkit-archive
在其中找到自己需要的CUDA版本,比如你之前在Tensorflow官网找