【cpu or gpu】【tensorflow】怎么查看用的是CPU还是GPU

方法1 

from tensorflow.python.client import device_lib

print(device_lib.list_local_devices())

　

参考博客

可用设备为：

[name: "/device:CPU:0"

device_type: "CPU"

memory_limit: 268435456

locality {

}]

原来只有一个CPU设备可用了。于是检查下tensorflow的版本情况：

pip3 list

各应用版本为：

tensorflow          1.10.1

tensorflow-gpu      1.9.0

方法2

1. 查看tensorflow-gpu是否存在
2. 查看tensorflow-gpu版本如果高于tensorflow容易默认启动gpu，因此尽量让gpu版本>=cpu

方法3

import torch

torch.cuda.is_available()

btw，使用GPU的方法：(tensorflow)

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"