本文主要是参考https://www.leiphone.com/news/201711/GCh0IBszXrxP1iHU.html 并在其基础上细化说明,以及排雷。
重要声明:
安装GPU版本的TensorFlow最重要的就是各种软件之间版本匹配,并非最新的软件最好,太新太旧都不行。强烈建议走别人已经走过的版本搭配,如果使用自己搭配的版本,需要之前自行确定是否兼容。如果懒,那就严格走别人的路。
需要准备的软件安装包
64位 Windows 10+ Anaconda (Python 3.5 )+ Visual Studio 2015 + CUDA 9.2 + CuDnn 7.1.4 for CUDA 9.2+ TensorFlow 1.8.0+pycharm
STEP 1: Anaconda 下载安装
Anaconda 的下载地址。注意下载最新版本的Anaconda 然后要将python的版本搞成3.5.具体操作见下文。
选择符合电脑操作平台Anaconda 的版本下载后安装。一切按照默认安装除了下图,需要添加环境变量的路径,因此要打勾。