tensorflow 使用CPU而不使用GPU的问题解决
tensorflow 使用CPU而不使用GPU的问题解决
粘贴自:https://www.cnblogs.com/hutao722/p/9583214.html
今天发现一个怪现象,在训练keras时,发现不使用GPU进行计算,而是采用CPU进行计算,导致计算速度很慢。
用如下代码可检测tensorflow的能使用设备情况:
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
可用设备为:
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}]</