专栏首页计算机视觉failed to allocate **M (** bytes) from device: CUDA_ERROR_OUT_OF_MEMORY,错误原因及解决方案

failed to allocate **M (** bytes) from device: CUDA_ERROR_OUT_OF_MEMORY,错误原因及解决方案

错误原因:GPU资源占用太大

config = tf.ConfigProto(allow_soft_placement=True) gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) config.gpu_options.allow_growth = True

找到如下红的的这句话,在这之前加上如上三行代码,在session前约束占用空间。能够使得tensorflow占用资源降至70%,当然也可以改为50%,而不是整个服务器GPU

with tf.Session() as sess: 将其改为 with tf.Session(config=config) as sess:

最终改为如下结构就可以啦

config = tf.ConfigProto(allow_soft_placement=True) gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.7) config.gpu_options.allow_growth = True with tf.Session(config=config) as sess:

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • Image-to-Image Translation with Conditional Adversarial Networks超细致解析:使用条件Gan经行图像的转换

    像素到像素的预测     每次写一篇都为了自己更深的思考,也希望大家能读懂...

    水球喵子
  • 为tensorflow指定GPU

    为tensorflow指定GPU,原因是,默认创建session时,会将所有显存占满,发现有人在用的时候,就会session不能创建而报错。 首先nvidia...

    水球喵子
  • linux 普通操作,查看资源使用情况

    nvidia-smi 查看当前资源 top-i 查看端口资源使用情况 进入命令行后直接输入即可~要讲的这么露骨吗

    水球喵子
  • 升级为私有全栈云的腾讯云TStack究竟强在了哪儿?

    在刚刚结束的美国丹佛Open Infrastructure峰会上,腾讯云对全新升级为私有全栈云的TStack进行了详细的介绍,引起峰会现场众多OpenStack...

    腾讯技术工程官方号
  • 升级为私有全栈云的腾讯云TStack究竟强在了哪儿?

    戳蓝字“CSDN云计算”关注我们哦! ? 在刚刚结束的美国丹佛Open Infrastructure峰会上,腾讯云对全新升级为私有全栈云的TStack进行了详...

    腾讯云TStack
  • 【tensorflow2.0】数据管道dataset

    如果需要训练的数据大小不大,例如不到1G,那么可以直接全部读入内存中进行训练,这样一般效率最高。

    绝命生
  • tensorflow 2.0模式下训练的模型转成 tf1.x 版本的pb模型实例

    升级到tf 2.0后, 训练的模型想转成1.x版本的.pb模型, 但之前提供的通过ckpt转pb模型的方法都不可用(因为保存的ckpt不再有.meta)文件, ...

    砸漏
  • tendermint-peers讲解

    Peer Exchange Reactor (PEX)确保与好的peers进行连接和通信

    _simple
  • Registrator中文文档

    Registrator监控新建的Docker容器,并且检查判定这些容器提供的服务。从我们的目的出发,任何监听在某个端口的程序都是服务。Registrator发现...

    少羽大怪兽
  • Django—第三方引用

    借助富文本编辑器,网站的编辑人员能够像使用offfice一样编写出漂亮的、所见即所得的页面。此处以tinymce为例,其它富文本编辑器的使用也是类似的。

    py3study

扫码关注云+社区

领取腾讯云代金券