首页 > 其他 > 详细

限制TensorFlow只在CPU上运行的方法

时间:2018-11-04 00:11:44      阅读:134      评论:0      收藏:0      [点我收藏+]

我们知道安装tensorflow可以安装两个版本的,有cpu和gpu的,如果同时安装的时候会默认进行用gpu进行,那如歌更改为使用cpu呢

在import tensorflow/keras 之前,加入如下代码

import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"  
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

 

限制TensorFlow只在CPU上运行的方法

原文:https://www.cnblogs.com/yanshuai0922/p/9902417.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!