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

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

这里写图片描述

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

  1. 下载 CUDA 和 cuDNN 安装包
    Google 官方推荐CUDA8.0和cuDNN v6.1,可以去各自官网下载。cudacuDNN
    这里写图片描述

  2. 安装CUDA
    直接双击左边exe安装,按提示下一步装到底就行。装完测试一下是否安装成功,命令行输入 nvcc -V ,看到版本信息就表示安装成功了。
    这里写图片描述
    这里写图片描述

  3. 安装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

  4. 安装TensorFlow
    安装部分看下篇帖子
    http://blog.csdn.net/lwplwf/article/details/54896088

  • 6
    点赞
  • 17
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值