虽然相关的文章有很多,但是我还是想记录一下我在windows 下安装 tensorflow 的全过程
一 安装python Anaconda
Anaconda 是一种python语言的免费增值开源发行版 ,用于进行大规模数据处理, 预测分析, 和科学计算, 致力于简化包的管理和部署。因为TensorFlow好像只支持python3.5 版本所以我直接下的是 Anaconda 的python 3.5 版本 Anaconda 4.2.0. ,但是官网推荐直接下最新版本然后内建3.5的环境,下载链接 Anaconda installer archive
下载后直接下一步安装 但是要将Anaconda 的路径加到系统路径
二 安装tensorflow
如果你的电脑有安装gpu,并且他的Compute Capability >=3 (英伟达gpu Compute Capability 一览表CUDA GPUs | NVIDIA Developer )就可以安装tensorflow的gpu版本了
有两种方式 一种是在系统的cmd 安装tensorflow 一种是在 Anaconda prompt 安装
我是在cmd命令行安装 pip install --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-0.12.0-cp35-cp35m-win_amd64.whl 按照指引执行命令即可
或者在 Anaconda prompt 中执行 以上命令 在Anaconda prompt 中安装再安装pycharm 会比较方便的编写Python程序,在cmd中安装似乎就是安装在公共目录下,ptyhon 环境读不到
三 安装cuba
我下载的是cuba 8.0 CUDA Toolkit 8.0 - Feb 2017 | NVIDIA Developer,按照指引点下一步安装,我也没有下载vs 就直接跳过了,可以使用,要注意的是要把Patch也下载,在base 之后安装就可以了
之后安装cudnn Membership Required | NVIDIA Developer 下载与cuba匹配的版本之后解压,我下的是5.1 版本的,之前下了个7.x的版本结果跑起来说动态库找不到,但是5.1 就可以运行,把bin目录加到系统的PATH路径下就可以了
四 测试
import tensorflowas tf
#import numpy as np
def main():
h1 = tf.constant([1, 2, 3, 4, 5])
h2 = tf.constant([6, 7, 8, 9, 10])
result = tf.multiply(h1, h2)
s = tf.Session()
print(s.run(result))
s.close()
main()
执行以上代码
以上,有什么不足或者错误之处欢迎大家指正