之前写了一篇是在 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