随笔分类 -  TensorFlow

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 做个读书人 阅读(3199) 评论(0) 推荐(0) 编辑
tf.test.is_gpu_available() 返回结果为False解决办法
摘要：安装完gpu版本的tensorflow,导入正常，但是tf.test.is_gpu_available()一直返回False,解决办法： 1、打开NVIDIA控制面板，查看CUDA的驱动版本，如果版本太低，到Nvidia Experience更新驱动程序。默认安装即可。（安装过程自行百度） 2、如果 阅读全文
posted @ 2021-03-09 18:32 做个读书人 阅读(3384) 评论(0) 推荐(0) 编辑