训练数据时候发现跑的很慢,查看一下是使用什么跑的
from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
输出:
[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 6321505847024071618
]
原来是cpu在跑,怪不得这么慢!