win10安装Tensorflow2.0GPU版本

前言

看到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

win10安装Tensorflow2.0GPU版本

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

 

win10安装Tensorflow2.0GPU版本

上一篇:如何将Linux的工程原封不动地移植到Windows上面


下一篇:kubernetes使用http rest api访问集群之使用postman工具访问 apiserver