GPU 版tensorflow在训练CIFAR数据集时GPU 没有参与计算/出现假死机状态/CPU满负荷运转

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/wangxinsheng0901/article/details/80681027

代码:《TensorFlow实战》黄文坚著,第5.3节 TensorFlow 实现进阶的卷积网络,所用数据集为CIFAR-10

出现问题的电脑型号:thinkpad T470; 显卡: GeForce 940mx

问题描述:书中提供的代码之前在不同的台式机上运行都没有问题,后来在笔记本运行的时候发现CPU满负荷运转,而GPU没有参与计算。开始以为是显卡相关驱动没有安装好,在对显卡驱动进行了几番卸载和安装后,TensorFlow终于从CPU负载下的缓慢计算变为了彻底不计算。。。运行程序后,一直停留打印显卡信息那一步,后来才知道在调用CIFAR数据时默认是用的GPU,这回大大降低调用的速度,在调用CIFAR读取程序的时候增加with tf.device('/cpu:0'),即指定用cpu即可解决:

 

 

 

展开阅读全文

没有更多推荐了,返回首页