作为一名谷粉再加上最近研究深度学习,不得不说TensorFlow是一个很好的切入点,今天看了下官方教程踩了几个坑后,终于在PyCharm上起飞了,废话不多说下面开始搭建环境吧。(此教程Win10,8.1和7都适用,只需下载不同的安装文件即可)
1. 确定安装版本
CPU版:如果你的电脑没有NVIDIA的显卡的话,你就必须选择安装这个版本,不过这个版本的安装要比GPU版的简单,官方也推荐先用CPU版的来体验。
GPU版:TensorFlow在GPU上运行要比CPU上快很多,如果你的GPU能够达到要求就可以选择安装GPU版。
GPU版的安装要求
- CUDA® Toolkit 8.0
- CUDA® Toolkit 8.0的相关驱动(这个在安装程序里会带)
- cuDNN v5.1
- 确认你GPU的CUDA的计算能力高于3.0(NVIDIA官方支持的对照表这里)
2.安装准备
- CUDA® Toolkit 8.0
- cuDNN v5.1
- Anaconda 或者原生的pip
PyCharm(可选)
P.S:这里的Python版本一定要选3.5.X,写这篇博客的时候官方在Windows平台上只支持3.5.X。若选择了其他版本有可能会报* Could not find a version that satisfies the requirement tensorflow-gpu (from versions: ) No matching distribution found for tensorflow-gpu*之类的错误!
若已经选了Python的其他版本,现有的解决方案如下(Anaconda办法):
P.S:一行一行的运行
实际上这里是Anaconda多版本Python管理工具,详细见这里
conda create --name tensorflow python=3.5
activate tensorflow
conda install jupyter
conda install scipy
pip in