首页 > Windows开发 > 详细

win10安装Tensorflow2.0GPU版本

时间:2019-06-27 20:27:16      阅读:898      评论:0      收藏:0      [点我收藏+]

前言

看到DateWhale出了一篇安装教程(微信公众号DateWhale),决定体验一下Tensorflow2.0的GPU版本。。

正文

总共所需环境为

Anaconda3.X,

CUDA9.0以上,以及

cuDNN7.5以上的配置,

一、

首先Anaconda是已经有3.6的版本了,

二、

然后在自己机子上看了一下环境变量,原来很久以前我就装了CUDA的9.0版本,挺爽。。

测试一下CUDA是安装成功的,CMD打开命令行窗口,敲:

nvcc -V

#

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2017 NVIDIA Corporation
Built on Fri_Sep__1_21:08:32_Central_Daylight_Time_2017
Cuda compilation tools, release 9.0, V9.0.176

出现了CUDA的版本信息,我这里是9.0,那么就表示CUDA安装成功了。然后在CUDA的安装文件夹里C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0

在bin和include文件夹里找了一圈都没找到cuDNN这个库,说明我应该重新下载。。

三、

于是,在这个网站里下载和CUDA对应版本的cuDNN——https://developer.nvidia.com/rdp/cudnn-download

技术分享图片

可以看到我应该是需要下载7.6.1版本的,于是下载。。

 

win10安装Tensorflow2.0GPU版本

原文:https://www.cnblogs.com/yqpy/p/11099236.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!