查看 tensorflow 是GPU版本 还是CPU版本

在Python环境中输入：

import os

from tensorflow.python.client import device_lib

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"

if __name__ == "__main__":

    print(device_lib.list_local_devices())

输出：

[name: "/cpu:0"

device_type: "CPU"

memory_limit: 268435456

locality {

}

incarnation: 5641726301910989523

]

说明是CPU版本。