tensorflow查看电脑的CPU和GPU

1、查看电脑GPU和CPU 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()) 2.指定CPU或GPU进行计

>>阅读原文<<