GPU版TensorFlow设置CPU运行的办法

import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 将以上三行代码放入程序首部便可。import
相关文章
相关标签/搜索