GPU版TensorFlow怎么指定让CPU运行

因为某些缘由GPU版的TensorFlow运行起来会出现一些问题,好比内存溢出等状况。此时咱们能够用CPU和系统内存来运行咱们的程序。html 代码以下:post import os os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 将以上三行代码放入程序首部便可。 --
相关文章
相关标签/搜索