原因是在gpu下超出显存,不得已在cpu下训练模型
该链接中有,点击打开链接
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

要在import keras之前加上上述代码

kera tensorflow只使用cpu训练相关推荐

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

    笔记本是NVIDIA GeForce 940M的显卡,只有2G的显存,运行TensorFlow代码时候常出现OOM(Out of Memory)的错误,原因是batch_size设置得太大导致显存不足 ...

  2. tensorflow配置只使用CPU

    文章目录 1 方法一 2 方法二 有些场景下,比如GPU版本运行失败或其它原因,需要强制tensorflow使用CPU,这里提供两种方法,仅针对tensorflow2 1 方法一 import ten ...

  3. CNN卷积神经网络—LeNet原理以及tensorflow实现mnist手写体训练

    CNN卷积神经网络-LeNet原理以及tensorflow实现minst手写体训练 1. LeNet原理 2.tensorflow实现Mnist手写体识别 1.安装tensorflow 2.代码实现手 ...

  4. TensorFlow 强制使用CPU

    TensorFlow 强制使用CPU 因为显存只有8G,训练时各种报错.有时训练几次就崩了.所以直接使用cpu,有条件的还是使用gpu的快. import os os.environ["CU ...

  5. 用C++调用tensorflow在python下训练好的模型(centos7)

    本文主要参考博客https://blog.csdn.net/luoyexuge/article/details/80399265 [1]  bazel安装参考:https://blog.csdn.ne ...

  6. python编译环境对cpu要求高不高_解决Tensorflow 使用时cpu编译不支持警告的问题

    使用TensorFlow模块时,弹出错误Your CPU supports instructions that this TensorFlow binary was not compiled to u ...

  7. 使用tensorflow object detection API 训练自己的目标检测模型 (三)

    在上一篇博客"使用tensorflow object detection API 训练自己的目标检测模型 (二)"中介绍了如何使用LabelImg标记数据集,生成.xml文件,经过 ...

  8. TensorFlow 网络模型移植和训练指南

    TensorFlow 网络模型移植和训练指南(持续更新) 1.限制 tensorflow只兼容tensorflow1.15 3.网络迁移 3.1 使用 Estimator 迁移 关于估算器 Estim ...

  9. 【深度学习】基于tensorflow的服装图像分类训练(数据集:Fashion-MNIST)

    ​ 活动地址:CSDN21天学习挑战赛 目录 前言 了解Fashion-MNIST数据集 下载数据集 使用tensorflow下载(推荐) 数据集分类 数据集格式 采用CPU训练还是GPU训练 区别 ...

最新文章

  1. 游戏+云服务器+自动驾驶样样全,看NVIDIA为中国市场带来了哪些好东西 | GTC China 2018...
  2. spring Transaction Management --官方
  3. 996和被辞退,二选一
  4. Deepin 15.7安装教程
  5. python计算和差积商_Python与处理计算的区别
  6. 基于TerraExplorer Pro的校园三维浏览系统开发(转载)
  7. linux 邮件文件名 病毒,文件型File/Macro病毒的捕获
  8. java json.stringify_JSON.stringify() 方法
  9. 利用jszip压缩下载多文件
  10. 气象统计 第八次实习 EOF分析
  11. 分段概率密度矩估计_2017年考研《数学一》大纲
  12. 实验室计算机化系统操作规程,gmp附录:计算机化系统.pptx
  13. DTI及MRI数据预处理
  14. 为什么计算机技术要革命,从智慧联想的发布谈人工智能革命
  15. 计算机版本歌曲,酷我音乐2017旧版大全
  16. 云曦网络空间安全实验室第一次考试WP
  17. 代码随想录训练营day31
  18. swiper炫酷_Swiper走马灯
  19. 控件拖动后,某些事件引起的布局重置或位置还原问题
  20. Push Mail技术(转)

热门文章

  1. 我的新书《Flink大数据分析实战》出版啦
  2. 强推大杀器xgboost
  3. 南方CASS安装,打开不能正常运行怎么办?
  4. 吉大21c语言在线作业,21春吉大《脚本语言和互动网页设计》在线作业二参考
  5. 异形工厂自动化工厂教程
  6. window 查看公网IP地址
  7. Linux常用工具的安装
  8. 剑指Offer 10- II—青蛙跳台阶问题
  9. 国王挖金子(动态规划)
  10. matlab axes subplot,matlab编程如何在一个axes里面使用subplot