转,想用compact CNN训练开源数据,在只装了tensorflow cpu版本的pycharm上跑,报该错,上网查的结果:
- 一般是由于模型是用GPU训练出来的,数据格式为NCHW,但是你用的tensorflow-cpu版本,仅支持NHWC格式。所以安装tensorflow-gpu版本就行。
- 如果安了tensorflow-gpu版本还是失败。看下运行log是否使用GPU。如下表示未使用到GPU,还是用的CPU在跑。xiang
failed call to cuInit: CUDA_ERROR_NO_DEVICE