首页 > 其他分享 >tensorflow:OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/re

tensorflow:OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/re

时间:2022-10-26 21:35:32浏览次数:99  
标签:tensor when job shape tensorflow allocating 256


运行TensorFlow(GPU)报错如下:

OOM when allocating tensor with shape[225,256,256,36] and type float on /job:localhost/replica:0/tas

原因是GPU OOM内存不够,因此可尝试改成批处理。即将训练的数据块调小。

参考链接:
​​​https://stackoverflow.com/questions/39076388/tensorflow-deep-mnist-resource-exhausted-oom-when-allocating-tensor-with-shape​

​https://github.com/tensorflow/tensorflow/issues/609​


标签:tensor,when,job,shape,tensorflow,allocating,256
From: https://blog.51cto.com/u_15847885/5798577

相关文章

  • prettytensor 卷积神经网络mnist识别例子
    代码importprettytensorasptimporttimefromdatetimeimporttimedeltaimportmatplotlib.pyplotaspltimporttensorflowastfimportnumpyasnpfromsklearn.metri......
  • 使用 oceanbase做为xxljob 以及nacos 的mysql存储
    如果我们的xxljob以及nacos集群比较多,同时需要提供ha以及灵活的mysql管理能力,基于tidb以及oceanbase是一个不错的选择oceanbase多租户能力上是比较方便的,可以更好......
  • POJ 3256(SPFA)
    这题只能对每一个点查一遍……有向图的话能用floyd,可是迫于时限用了SPFA。Programaa;constmaxk=10000;maxn=10000;maxm=10000;vark,n,m,i,j,l:longint;a:ar......
  • bladex-xxl-job使用
    1.整合bladex,配置docker,install 后编辑DockerFile.. build..push到HaBor,并且安装jdknacossentinelredisminio等等组件,这里只展示dockercomposeyml内容vers......
  • tensorflow中session的用法,莫烦的例子(2)
    importtensorflowastf#创建两个矩阵maxtrix,然后输出两个矩阵相乘的结果matrix1=tf.constant([[3,3]])#constant表示maxtrix1是一个常数,[3,3]表示是一个一行两列的......
  • 牛客SQL-牛客题:256-288(不干了)
    256.写一个sql查询,积分表里面出现三次以及三次以上的积分。若有多个符合条件的number,则按number升序排序输出SELECT`number`FROMgradeGROUPBY`number`HAVINGCOUN......
  • Tensorflow Lite从入门到精通
    TensorFlowLite是TensorFlow在移动和IoT等边缘设备端的解决方案,提供了Java、Python和C++API库,可以运行在Android、iOS和RaspberryPi等设备上。目前TF......
  • 按照tensorflow jupyer
    anaconda 安装包 链接:https://pan.baidu.com/s/1wh6mYu1uMLlPM5Ai5ONCXQ提取码:3rlo 两个库pipinstalltensorflow==2.8.0keras==2.8-ihttps://pypi.douban.co......
  • Tensorflow解析tfrecord
    1、序列化#coding:utf-8from__future__importabsolute_importfrom__future__importdivisionfrom__future__importprint_functionimportnumpyasnpimpor......
  • win11+cuda11.2+cudnn+Tensorflow-GPU 环境配置
    名词解释CUDA即英伟达的显卡并行计算框架,nvidia-smi可以查看tensorflow-gpu的运行需要它的底层支持,它是一个计算框架,抽象层次比驱动高,每个版本的CUDA都是基于一定版......