前言
看到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版本的,于是下载。。