2021年3月11日
TensorFlow使用GPU训练时CPU占用率100%而GPU占用率很低
摘要： 在训练keras时，发现不使用GPU进行计算，而是采用CPU进行计算，导致计算速度很慢。 用如下代码可检测tensorflow的能使用设备情况： from tensorflow.python.client import device_lib print(device_lib.list_local_d 阅读全文
posted @ 2021-03-11 18:09 做个读书人 阅读(3263) 评论(0) 推荐(0)