GPU版TensorFlow設置CPU運行的辦法

import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 將以上三行代碼放入程序首部便可。import
相關文章
相關標籤/搜索