Windows 10 安装 1080 ti GPU版本TensorFlow 踩坑记录

时间:2024-03-28 17:25:06

本文主要是参考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自带驱动版本高,都可能出现这个问题)

Windows 10 安装 1080 ti GPU版本TensorFlow 踩坑记录

解决方案:

直接点击继续—>同意并继续—>自定义(高级)—>只选择CUDA进行安装,最后安装成功

注意最后一个自定义选项操作,后面两个不打勾:

Windows 10 安装 1080 ti GPU版本TensorFlow 踩坑记录

2.  TensorFlow必须是安装 1.3,而不能使用默认的安装命令,否者会默认下载最新的TensorFlow。

即,第七步 使用的命令是: pip3 install tensorflow-gpu=1.3 

如果之前已经下载错了TensorFlow,只需要用  pip3 unstall tensorflow-gpu 卸载即可,然后重新使用 pip3 install tensorflow-gpu=1.3 即可。