TensorFlow安装--（方法2）GPU环境配置部分（装CPU版跳过不用看）（Windows10 64位）

最新推荐文章于 2025-08-27 11:48:35 发布

原创最新推荐文章于 2025-08-27 11:48:35 发布 · 2.4w 阅读

CC 4.0 BY-SA版权

文章标签：

81 篇文章

订阅专栏

20 篇文章

订阅专栏

16 篇文章

订阅专栏

本文详细介绍了在Win10 64位系统上安装GPU版本TensorFlow的过程，包括所需环境配置如CUDA和cuDNN的安装步骤及验证方法。

之前写了一篇是在 ubuntu 14.04下装 TensorFlow，装的是cpu版本的，又试着装gpu的但是出现了很多蜜汁问题，担心把之前的caffe环境破坏了，既然现在官方支持Windows系统了，就试试在win10 64位下安装 TensorFlow。

TensorFlow 有cpu和 gpu两个版本。gpu版本需要英伟达CUDA 和 cuDNN 的支持，cpu版本不需要。

这里我装的是gpu版

环境：

win10 64位
gpu：gtx1080Ti（确认你的显卡支持 CUDA，这里确认）
确认你安装了VS2015或者2013或者2010（我用的2015）
python 3.5（确认你的Python版本是3.5 64位。关于python建议安装Anaconda，因为这个集成了很多科学计算所必需的库，能够避免很多依赖问题，安装也特别方便。这里下载）
pip 9.0.1（确认pip版本 >= 8.1，用pip -V 查看当前 pip 版本，用python -m pip install -U pip升级pip 。）

这里写图片描述

准备工作完成后就可以开始了：

下载 CUDA 和 cuDNN 安装包
Google 官方推荐CUDA8.0和cuDNN v6.1，可以去各自官网下载。cuda；cuDNN；
安装CUDA
直接双击左边exe安装，按提示下一步装到底就行。装完测试一下是否安装成功，命令行输入 nvcc -V ，看到版本信息就表示安装成功了。
安装cuDNN
解压zip压缩文件，放到任何一个目录下就行，然后把放的那个目录添加到Path环境变量里。

然后将刚才解压出来的下面这些文件复制到相应位置，我解压在了E盘根目录下了。

E:\cuda\bin\cudnn64_6.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
E:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
E:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64
安装TensorFlow
安装部分看下篇帖子
http://blog.csdn.net/lwplwf/article/details/54896088