准备工作:
- 支持cuda的显卡(使用tensorflow GPU版本时)
- Anaconda3安装包
- cuda和cuDNN安装包
安装过程
1. 安装cuda
https://developer.nvidia.com/cuda-toolkit-archive
CUDA是由显卡厂商NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
现在CUDA最新版是9.1
我安装的tensorflow 1.7.0要求cuda为9.0版的。所以根据tensorflow的要求下载对应的cuda就行。
下载完后直接双击安装下一步即可,默认条件下环境变量都已经添加好,可以进入查询
下载cuDNN,解压安装包,并将包内的bin,lib,include等的地址添加到环境变量中
2. 安装Anaconda
Anaconda是由Python提供支持的领先的开放数据科学平台。 Anaconda的开源版本是Python和R的高性能分发版本,包括超过100个用于数据科学的最流行的Python,R和Scala软件包。
地址:
https://www.anaconda.com/download/#windows
最新的anaconda默认是3.6的python。
问题关键是tensorflow要求3.5.2版本的python,而现在安装最新的anaconda默认是3.6的python。
解决方法如下:
管理员模式打开cmd
输入conda install python=3.5.2
然后再正常pip安装tensorflow即可
此时spyder和ipython也是python 3.5.2
安装anaconda时记得勾选上地址添加到环境变量
包括网络结构
3. 安装tensorflow
由于 Google 那帮人已经把 tensorFlow 打成了一个 pip 安装包,所以现在可以用正常安装包的方式安装 tensorFlow 了,就是进入命令行执行下面这一条简单的语句
# GPU版本
pip install --upgrade tensorflow-gpu
# CPU版本
pip3 install --upgrade tensorflow
为tensorflow指定编译器:anaconda目录下的python.exe
或者通过利用conda安装python3.5
Conda create --name Tensorflow(你的环境名称) python=3.5(你的python版本号)(用来创建你自己的环境,特别是适用于多人共用服务器时使用)
然后再activate Tensorflow,然后输入安装tensorflow的命令进行安装
此时,因为python和tensorflow都是在conda的环境下安装的,所以指定编译器时要指定conda环境下的python.exe
也就是anaconda/envs/tensorflow/python.exe