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

时间:2024-01-14 11:18:26

在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版本。