尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
【Tensorflow实战系列】Tensorflow: CUDA_ERROR_OUT_OF_MEMORY
最近LZ一直在使用tensorflow,遇到有些小问题,就遇到一个,解决一个,有些一直重复出现,或者是做个笔记,加快小伙伴的使用.在写测试程序代码的时候出现如下报错,源代码就不贴出来了…2018-11-07 17:10:30.875135: E tensorflow/stream_executor/cuda/cuda_driver.cc:924] failed to allocate 62.25M (65273856 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY看看好像怎么连60多兆的空间GPU都分配不出来了,那么碰到这个问题怎么解决呢?在定义session前,加上一些设置,使得程序更加完整#tf.ConfigProto函数在创建session的时候,用来对session进行参数配置configtf.ConfigProto()#允许tf动态的申请现存config.gpu_options.allow_growthTrue#或者限制GPU使用率config.gpu_options.per_process_gpu_memory_fraction0.4#允许tf自动选择一个存在并且可用的设备来运行操作config.allow_soft_placementTrue#获取operations和tensor被指派到哪个设备(几号cpu或者gpu)#会在终端打印出各项操作是在哪个设备上运行#一般设成Falseconfig.log_device_placementFalse#最后将设置传给session中,再次运行代码,OK,问题解决!sesstf.Session(configconfig)代码之路且行且珍惜!小伙伴们今天也要加油( ⊙ o ⊙ )
RELATED

相关推荐

【点云】python: 关于点云的一些基本处理

【点云】python: 关于点云的一些基本处理

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

📅 2026/10/22 1:01:28
【点云】Pointet的读后感(CVPR2016):1.相关工作的介绍

【点云】Pointet的读后感(CVPR2016):1.相关工作的介绍

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

📅 2026/10/22 1:01:28
【Tensorflow实战系列】Tensorflow:TypeError: pred must not be a Python bool

【Tensorflow实战系列】Tensorflow:TypeError: pred must not be a Python bool

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

📅 2026/10/22 1:01:28
MORE NEWS

更多资讯

📰

Gitea 版本控制服务端【Red Hat 离线安装】

官网 官方网站:https://about.gitea.cn/ 官方文档:https://docs.gitea.cn/ 官方下载地址:https://dl.gitea.com/gitea/ 各版本控制软件对比 对比项SVNGitMercurialPerforce版本控制类型集中式(依赖中央服务器)分布式…

📰

make_tuple 一文讲透:CANN 昇腾算子开发里最省心的元组创建方式

make_tuple 一文讲透:CANN 昇腾算子开发里最省心的元组创建方式 【免费下载链接】asc-devkit 本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C标准规范,主要由类库和语言扩展层构成,提供多层级API&#xff0…

📰

lxmusic--听歌自由,感谢开源

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

📰

读懂PIIP-LLaVA的trainer_state与training_args:5197步微调训练全复盘

【免费下载链接】PIIP-LLaVA_ConvNeXt-B_CLIP-L_1024-336_13B 项目地址: https://ai.gitcode.com/OpenGVLab/PIIP-LLaVA_ConvNeXt-B_CLIP-L_1024-336_13B 点击查看 免费下载 PIIP-LLaVA_ConvNeXt-B_CLIP-L_1024-336_13B 是基于 Vicuna-13B 的图像多模态大模型&…

📰

如何可视化TIPSv1-So400m/14的1024个空间Patch特征:PCA可视化实战

【免费下载链接】tipsv1-so400m14 项目地址: https://ai.gitcode.com/hf_mirrors/google/tipsv1-so400m14 点击查看 免费下载 TIPSv1-So400m/14(TIPS,ICLR 2025 提出的对比式视觉语言模型)最大的亮点之一,就是它能输出…

📰

无名杀完全上手指南:在浏览器里跑一套可自由改造的三国杀引擎

无名杀完全上手指南:在浏览器里跑一套可自由改造的三国杀引擎 【免费下载链接】noname 项目地址: https://gitcode.com/GitHub_Trending/no/noname 我把仓库克隆下来挂到本地服务器上,用 Chrome 内核浏览器打开页面,就能选武将、发牌…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

读完文章,想聊聊您的网站?

告诉我们您的行业与需求,资深顾问一对一梳理方案与报价,全程免费。

📞 💬