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進行計

>>阅读原文<<