最近买了个坠机堡垒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,测了一下也能跑了。
网友评论