【软件安装】根据Tensorflow-GPU安装CUDA+CUDNN

目录

1 版本关系

2 如何查看现有CUDA版本号

3 下载安装CUDA

4 安装下载CUDNN

         参考


背景:

win10要安装TensorFlow-GPU 1.4.0,现有CUDA为10.0,版本不符

问题:

ImportError: 
    Could not find 'cudart64_80.dll'. 
    TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. 
    Download and install CUDA 8.0 from this URL: https://developer.nvidia.com/cuda-toolkit

读题,是TensorFlow-GPU和CUDA版本不符,要CUDA8.0。

ImportError: 
    Could not find 'cudnn64_6.dll'. 
    TensorFlow requires that this DLL be installed in a directory that is named in your %PATH% environment variable. 
    Note that installing cuDNN is a separate step from installing CUDA, 
    and this DLL is often found in a different directory from the CUDA DLLs. 
    You may install the necessary DLL by downloading cuDNN 6 from this URL: https://developer.nvidia.com/cudnn

读题,是TensorFlow-GPU和CUDNN版本不符,要CUDNN6.0。

解决:

1 版本关系

要安装TensorFlow-GPU,要注意TF-GPU和CUDA、CUDNN的版本关系:(其实仔细看问题,就可以知道要下什么版本的了)

CUDA现有版本10.1,所以要去官网重装(版本太高不行的原因,可能是锅配盖,合适就行)。

2 如何查看现有CUDA版本号

推荐几个方法

Windows系统查看CUDA版本号(也可以在右下角找到NVDIA的标志,右键即可)

win10 查看安装的cuda版本号

3 下载安装CUDA

3.1 下载

如果仔细看问题1,你就可以看到后面附赠一个URL,很贴心。

附赠一个官网下载链接:CUDA Toolkit Archive,点进去,找合适自己的版本就可以了。

根据自己的系统选择,没毛病。

Installer Type安装类型可以选择网络安装还是本地安装,个人建议本地。

3.2 安装

自定义推荐都可以吧,忍不住点个自定义瞅瞅

安装完毕,检测下版本号

在cmd也可以

4 安装下载CUDNN

介绍:深度神经网络库(cuDNN)是GPU加速的用于深度神经网络的原语库。cuDNN为标准例程提供了高度优化的实现,例如向前和向后卷积,池化,规范化和激活层。

如果能打开这个页面:cuDNN Archive,下载对应版本,可以跳过4.1\4.2,从4.3开始。(能少走点弯路就少走一点)

4.1 前置步骤

又可以在问题2中找到官网URL:NVIDIA cuDNN

有点麻烦,他要求注册登录,还要写调查问卷。

(耐心点,可以用微信、QQ,但还是要绑定用户)

邮箱验证

填写细节

居然还有调查问卷!写写写

4.2 下载

终于到了下载页面,先打勾✔,发现版本不是我要的,点最下面的红框里的内容。

网址:https://developer.nvidia.com/rdp/cudnn-download

https://developer.nvidia.com/rdp/cudnn-archive

选择合适版本下载

4.3 安装

先打开之前安装的cuda的文件,默认位置是在 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA

如同,将CUDNN 压缩包中的文件,移动到相同文件名下。

4.4 环境配置

参考了三篇博文就有三种配置,先选了最简单的试试

往path里面加一条(注意版本号)

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

最简单的就搞定了。

 

参考

查看版本号

Tensorflow不同版本要求与CUDA及CUDNN版本对应关系

tensorflow-gpu版本与CUDA,Cudnn兼容问题,及GPU 查阅技巧

CUDA+CUDNN的教程

win10下CUDA和CUDNN的安装(超详细)!亲测有效

Windows10系统下的tensorflow-gpu+CUDA+cuDNN配置

Win10安装CUDA10和cuDNN
 

  • 3
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值