欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

根据Tensorflow-GPU安装CUDA+CUDNN

程序员文章站 2022-06-01 18:35:29
...

背景:

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的版本关系:(其实仔细看问题,就可以知道要下什么版本的了)

根据Tensorflow-GPU安装CUDA+CUDNN

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

2 如何查看现有CUDA版本号

推荐几个方法

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

win10 查看安装的cuda版本号

3 下载安装CUDA

3.1 下载

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

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

根据Tensorflow-GPU安装CUDA+CUDNN

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

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

根据Tensorflow-GPU安装CUDA+CUDNN

3.2 安装

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

根据Tensorflow-GPU安装CUDA+CUDNN

根据Tensorflow-GPU安装CUDA+CUDNN

根据Tensorflow-GPU安装CUDA+CUDNN

安装完毕,检测下版本号

根据Tensorflow-GPU安装CUDA+CUDNN

根据Tensorflow-GPU安装CUDA+CUDNN

在cmd也可以

根据Tensorflow-GPU安装CUDA+CUDNN

4 安装下载CUDNN

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

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

4.1 前置步骤

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

根据Tensorflow-GPU安装CUDA+CUDNN

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

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

根据Tensorflow-GPU安装CUDA+CUDNN

邮箱验证

根据Tensorflow-GPU安装CUDA+CUDNN

填写细节

根据Tensorflow-GPU安装CUDA+CUDNN

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

根据Tensorflow-GPU安装CUDA+CUDNN

4.2 下载

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

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

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

根据Tensorflow-GPU安装CUDA+CUDNN

选择合适版本下载

根据Tensorflow-GPU安装CUDA+CUDNN

4.3 安装

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

根据Tensorflow-GPU安装CUDA+CUDNN

如同,将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
 

相关标签: 工具使用说明