码迷,mamicode.com
首页 > 其他好文 > 详细

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

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

标签:ras   order   style   col   方法   import   dev   _id   使用   

我们知道安装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上运行的方法

标签:ras   order   style   col   方法   import   dev   _id   使用   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!