Win10下安装Tensorflow GPU版本记录

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/m_buddy/article/details/81008966

1. 前言

这里我使用的环境是Win10家庭中文版,显卡是GTX1050。由于安装Tensorflow相对安装Caffe来说比较简单,这里就直接给出安装步骤了。

1. 安装CUDA8

安装CUDA需要到这里(https://developer.nvidia.com/cuda-toolkit-archive)去下载,但是呢!需要注意一点的是CUDA的版本和cuDNN的版本要对应上,要不是会出问题。这里我选择的是CUDA8,下载之后就是安装了安装和一般的exe没啥区别,这里就过…

2. 安装cuDNN6

cuDNN是专用于神经网络计算加速的,其实他就是一包库文件,每一个CUDA版本都有特定与之匹配的cuDNN库文件,如CUDA 8.0就匹配cudnn-8.0-windows10-x64-v6,通常我们到官网上下载并解压缩后,直接将其文件替换到CUDA相应目录下即可!如下,解压缩cuDNN后在cuda目录下得到三个文件夹,里面分别有cudnn64_6.dll、cudnn.h及cudnn.lib。
这里写图片描述
我们将其拷贝合并到CUDA的安装目录C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0下
这里写图片描述
之后便是设置环境变量了,环境变量设置如下:
这里写图片描述

3. 安装Tensorflow

这里看了很多资料说要装什么Anaconda,这里就直接pip3了。不整那么麻烦了。

pip3 install tensorflow-gpu==1.4

这里指定了版本是因为当前安装的CUDA与cuDNN只支持哪个版本的,所以只能安装这个版本的了。若是需要安装1.5及其以上的版本就需要安装CUDA9与对应的cuDNN了。
接下来就可以开始使用Tensorflow了。

猜你喜欢

转载自blog.csdn.net/m_buddy/article/details/81008966