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

Keras强制使用CPU

时间:2019-04-22 12:39:39      阅读:270      评论:0      收藏:0      [点我收藏+]

标签:导入   使用   问题   ice   ble   训练   dev   div   代码   

最近使用GPU来跑Keras模型速度很快,但是预测的时候加载的非常慢,估计是使用GPU的问题

GPU做并行运算效果优于CPU,但是在预测的时候不需要并行运算,如果再使用GPU的话导致加载时间太长

因此训练模型使用GPU,预测使用CPU

 

在导入Keras之前加入以下两行代码:

import os

os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

既可以使用CPU来预测模型

Keras强制使用CPU

标签:导入   使用   问题   ice   ble   训练   dev   div   代码   

原文地址:https://www.cnblogs.com/yqs-0705/p/10749113.html

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