GPU版TensorFlow怎么指定让CPU运行

由于某些原因GPU版的TensorFlow运行起来会出现一些问题,比如内存溢出等情况。此时我们可以用CPU和系统内存来运行我们的程序。

代码如下:

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
将以上三行代码放入程序首部即可。
---------------------
作者:MarTin Guo
来源:CSDN
原文:https://blog.csdn.net/Geoffrey_MT/article/details/81197030
版权声明:本文为博主原创文章,转载请附上博文链接!

posted @ 2019-07-22 14:36  交流_QQ_2240410488  阅读(1430)  评论(0编辑  收藏  举报