tensorflow-gpu安装遇到的坑

最近买了个坠机堡垒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安装遇到的坑)