Anaconda TensorFlow安装(Windows10 64位 gpu)

来源:互联网 发布:淘宝运营员是做什么的 编辑:程序博客网 时间:2024/06/04 19:15

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

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

这里我装的是gpu版

环境: 
- win10 64位 
- gpu:gtx960(确认你的显卡支持 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 和 cuDNN5,可以去各自官网下载。cuda;cuDNN;或者直接从我百度云拿,密码:4d9s
  2. 安装CUDA 
    直接双击左边exe安装,按提示下一步装到底就行。装完测试一下是否安装成功,命令行输入 nvcc -V ,看到版本信息就表示安装成功了。 
    这里写图片描述 
    这里写图片描述
  3. 安装cuDNN 
    解压zip压缩文件,放到任何一个目录下就行,然后把放的那个目录添加到Path环境变量里。 
    这里写图片描述 
    这里写图片描述
    然后将刚才解压出来的下面这些文件复制到相应位置,我解压在了E盘根目录下了。 
    这里写图片描述 
    E:\cuda\bin\cudnn64_5.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 
    打开命令提示符执行下面语句:

GPU版本

pip install tensorflow-gpu
  • 1
  • 1

CPU版本

pip install tensorflow
  • 1
  • 1

然后等安装完成 
这里写图片描述
然后命令行进 python 
import tensorflow as tf测试

下面是用一个简单的矩阵乘法测试一下,

a = tf.random_normal((100, 100))b = tf.random_normal((100, 500))c = tf.matmul(a, b)sess = tf.InteractiveSession()sess.run(c)
  • 1
  • 2
  • 3
  • 4
  • 5
  • 1
  • 2
  • 3
  • 4
  • 5

哇咔咔,现在可以愉快的玩耍了,不过,Windows下果然还是没有Linux下用着舒服啊。

补充:

  1. 更新TensorFlow
pip uninstall tensorflowpip install tensorflow
原创粉丝点击