本文主要是参考https://www.leiphone.com/news/201711/GCh0IBszXrxP1iHU.html 并在其基础上细化说明,以及排雷。
因此建议先看上述博文,然后看完本文再下载安装软件!
强烈建议,先装Anaconda,如果不装,之后导入pycharm时,解释器需要去找python3.5的安装位置,而不是Anaconda的虚拟环境位置。此外需要装scipy和其他小包会相对麻烦一些。
重要声明:
安装GPU版本的TensorFlow最重要的就是各种软件之间版本匹配,并非最新的软件最好,太新太旧都不行。强烈建议走别人已经走过的版本搭配,如果使用自己搭配的版本,需要之前自行确定是否兼容。如果懒,那就严格走别人的路。
需要准备的软件安装包
64位 Windows 7/8/10 或者Server 2012/2016 + Python 3.5 + Visual Studio 2015 + CUDA 8.0 + CuDnn 6.0+TensorFlow 1.3。
注意:截至本文的发表时间,上述的各个软件都有更新的版本,本人太懒,不想尝试更新的搭配,因此就这样搭配图省事,这种 搭配多人亲测可行。
下面是针对开头提到的博文的排雷
1. 安装CUDA时会出现“ could not find compatible graphic hardware”的警告,原因可能是CUDA 8.0 识别不了 1080 ti,或者本机的显卡驱动版本比CUDA8.0中自带的驱动版本高(实际上,不论CUDA装的哪个版本,只要本机驱动比CUDA自带驱动版本高,都可能出现这个问题)
解决方案:
直接点击继续—>同意并继续—>自定义(高级)—>只选择CUDA进行安装,最后安装成功
注意最后一个自定义选项操作,后面两个不打勾:
2. TensorFlow必须是安装 1.3,而不能使用默认的安装命令,否者会默认下载最新的TensorFlow。
即,第七步 使用的命令是: pip3 install tensorflow-gpu=1.3
如果之前已经下载错了TensorFlow,只需要用 pip3 unstall tensorflow-gpu 卸载即可,然后重新使用 pip3 install tensorflow-gpu=1.3 即可。