cuda nvcc版本不一致_Windows10安装tensorflow-gpu(2019.6新版本)

基本配置:Windows10家庭版+Anaconda虚拟环境Python3.6.8+Tensorflow-gpu-1.14.0+CUDA 10.0+cuDNN 7.4.2(1.13.0版本开始支持python3.7和CUDA 10、TensorRT 5.0)

667419d43e3638613a542a4713376af1.png

cd79df7c13d71b028b0cd775675e12bf.png

一、安装 Visual C++ 生成工具 2015(本人平时不怎么使用vs所以采取简洁安装)

说明:官方安装教程显示也是支持其他版本的:

ce3291039243f950172e05910a6be0df.png

tensorflow中文官网教程就是以VS2015参照的,而且还给出了经过测试的各版本组合(目前暂未给出win10+cuda10,本人想尝试安装比较新的版本)。

40bd62a4e84a9ccb363f2405f667459c.png

首先下载Visual C++ 生成工具 2015。此软件包随附在 Visual Studio 2015 中,但可以单独安装。进入Visual Studio 下载页面,选择“可再发行组件和生成工具”,下载并安装:
Microsoft Visual C++ 2015 Redistributable 更新 3,Microsoft 生成工具 2015 更新 3。

597b607b530b1593bfa9204e2f83a893.png

c5cfc08de885f43441a847e40ad4d688.png

二、安装CUDA 10.0

首先打开NVIDIA控制面板查看自己的显卡和驱动支持多少版本的CUDA,为了安装最新版本的CUDA我把显卡驱动升级到最新了,升级前可以找个软件备份下原驱动。

58870c5010abf4cdf7a965f475882e4b.png

2fb06e289ac0c562868eb6f3ee37c0b2.png

ec9c6713bedbc4be331632bf3ca22f8a.png

进入官网下载界面下载相应版本cuda。

40009b133ec6237c52db3279bb4286ba.png

f8d9dba7aad3a8c528fe21e83b7044b1.png

下载后的安装步骤参考官方教程。

ca59be151f8923131be0b65e25436e12.png

0f9392ad31bd4054ba2cb3e1d00dd462.png

9f6b5443f9eeed1a29de9c76d76bcd97.png

安装完成后简单检查一下cuda的版本:nvcc -V(更为完善的检查文末会提到)。

d20c06ceeba9993d4748b415a4998d41.png

再看一下环境变量是否配置成功,这个应该是在安装过程中自动添加了。右键“此电脑”-“属性”-“高级系统设置”-“高级”-“环境变量”。

aedcf8c02d318d5193a90df92fc7137c.png

842a4ea6acb299a191faa33ea5f1dbda.png

3acd3e0074e5d093fe72e03c1481131a.png

三、安装cuDNN7.4.2

进入官网下载界面下载对应cuda10.0的cudnn7.4.2(这里选择v7.4.2版本是考虑到下一个发布的7.5.0版本开始支持cuda10.1了,为规避安装失败的风险)。

1a9a86e458c650d378b1d2f9549afa4f.png

89fe75eac7513ab7d996703181de40ff.png

下载后的安装步骤参考官方教程。

解压cudnn7.4.2得到cuda文件夹,然后复制下面三个文件到 CUDA Toolkit 目录(CUDA我是默认目录安装的) 。

复制<cuDNN路径下>cudabincudnn64_7.dll 到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0bin.
复制 <cuDNN路径下>cuda includecudnn.h 到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0include.
复制 <cuDNN路径下>cudalibx64cudnn.lib 到 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0libx64.

6dfa291f636c9e1de664c9c0ed7707ad.png

按照官方说明还需对 cudnn.lib 进行链接,我没链接也能正常使用了(其实我不会哈哈)。

四、安装tensorflow-gpu并测试

pip install tensorflow-gpu

e015355775292ba6168fbc6c8dfc24b7.png

db37b071307505f89ba691ebd15461cf.png

简单测试tensorflow-gpu

import 

765d6959b8b8341a2643728d54fb8a81.png

五、补充说明踩过的坑

其实一开始下载的是CUDA最新版cuda_10.1.168_425.25_win10.exe,然后搭配的对应于cuda10.1的cuDNN最新版cudnn7.6.2,两者都已成功安装并通过比“nvcc -V”更进一步的测试:用VS2015打开C:ProgramDataNVIDIA CorporationCUDA Samplesv10.1(默认的安装目录)下的Samples_vs2015.sln文件,找到1_Utilities文件右键点击生成,生成完毕后去C:ProgramDataNVIDIA CorporationCUDA Samplesv10.1binwin64Release文件下cmd分别运行deviceQuery(测试能否找到显卡并且驱动是否正确)和bandwidthTest(测试CUDA版本及其它必要的测试),如下图:

380c6c9091eea5b45efab47fe21c53d1.png

7928ed6430825d03eeebae8dafdf2a09.png

54fbd41e568d9973d6baa4803857d8b6.png

0b6a0fa4e8e8b763f76c1269730491d3.png

75964bfbe8a7bf6f7ea1715fc3cc7238.png

但是后面 import tensorflow 的时候出现了如下错误:

71e0c0a08a5870691d284c2d860ebcfe.png

这个报错的意思应该是当前的tensorflow-1.14.0版本不支持CUDA-10.1,它提供了官方网址让去下载CUDA-10.0,另外尚未了解tensorflow-2.0新版本是否支持最新的CUDA-10.1。

PS:来回折腾了两天多可算给windows10配置了比较新版的环境,下一步打算在Ubuntu18.04上也尝试下新版本(据说直接 conda install tensorflow-gpu 一句搞定,不知道是不是配置的最新版本),后续如果踩坑比较多的话再追加。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值