tensorflow打印可用设备列表

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

猜你喜欢

转载自www.cnblogs.com/chibaojiushui/p/12364878.html