美文网首页
tensorflow-gpu安装遇到的坑

tensorflow-gpu安装遇到的坑

作者: Lnstark | 来源:发表于2018-12-09 18:36 被阅读0次

    最近买了个坠机堡垒6,gtx1050ti的,想装个gpu版的tensorflow。

    按网上csdn的教程依次下载cuda,cuDNN,最后安装tensorflow。然而从cuda8.0,9.0,9.2到10.0都试了一遍,tensorflow都跑不通。一开始以为是tensorflow的问题,也试了不同版本的tensorflow,然而都不行。后来觉得可能是cuda的问题,最后一次装的是cuda9.0,在vs2017里建了一个cuda项目,跑起来也有问题:

    一个是版本不支持问题 #error: -- unsupported Microsoft Visual Studio version!

    参考此问题下面Chao同学的回答解决。

    另一个是编译问题 

    Error expression must have a constant value C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Tools\MSVC\14.12.25827\include\type_traits 504

    参考此问题下面Wido Seidel童鞋的回答,安装VC++ 2017 version 15.4 v14.11 toolset并修改配置文件(路径可能得修改)得以解决。

    之后cuda项目能跑了,然后pip install tensorflow-gpu,测了一下也能跑了。

    相关文章

      网友评论

          本文标题:tensorflow-gpu安装遇到的坑

          本文链接:https://www.haomeiwen.com/subject/tzskhqtx.html