昨天在使用GPU服务器跑keras程序的时候提示这个错误,并在报完这个错误之后就直接吐核了。我用的是keras库,tensorflow backend。今天找到原因:tensorflow版本不正确。首先要下载tensorflow-GPU库:

conda install tensorflow-GPU==1.4.1

我们实验室的服务器安装的是CUDA8.0,适合的tensorflow版本为1.4.1,不同版本cuda会兼容不同版本tensorflow,如果不兼容,报错信息会提示,按照报错信息重新安装tensorflow-GPU就可以了。
好事多磨,又遇到TypeError: while_loop() got an unexpected keyword argument ‘maximum_iterations’
原因是keras的版本不兼容tensorflow-GPU1.4.1的版本,因此强制安装低版本keras:

conda install keras==2.1.2

就可以了。

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐