【tensorflow】查看所使用的tensorflow是GPU仍是CPU版本

激活tensorflow環境,在python環境中輸入: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_devic
相關文章
相關標籤/搜索