六级有手就行 2022-08-12 17:04 采纳率: 0%
浏览 23

keras和tensorflow时加载模型后服务器显存被吃满

在利用keras和tensorflow时加载模型后服务器显存几乎被吃满,试了网络上的方法,包括动态分配显存和设置固定显存,都没有效果,是我忽略了什么细节吗?

config = tf.ConfigProto()
config.gpu_options.allow_growth=True #不全部占满显存, 按需分配
sess = tf.Session(config=config)

import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
from keras.backend.tensorflow_backend import set_session
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
set_session(tf.Session(config=config))
这两种代码都试过没有效果,我用的tensorflow-gpu==1.14.0,keras==2.2.4,cuda10.1,在加载模型时占用的显存也挺多的,各位有无什么意见

  • 写回答

1条回答 默认 最新

  • kakaccys 2022-08-12 19:52
    关注

    tensorflow-gpu1.14.0对应的cuda是10.0,不知道楼主cuda10.1怎么安装成功的,估计和这有点关系

    评论 编辑记录

报告相同问题?

问题事件

  • 创建了问题 8月12日

悬赏问题

  • ¥15 is not in the mmseg::model registry。报错,模型注册表找不到自定义模块。
  • ¥15 安装quartus II18.1时弹出此error,怎么解决?
  • ¥15 keil官网下载psn序列号在哪
  • ¥15 想用adb命令做一个通话软件,播放录音
  • ¥30 Pytorch深度学习服务器跑不通问题解决?
  • ¥15 部分客户订单定位有误的问题
  • ¥15 如何在maya程序中利用python编写领子和褶裥的模型的方法
  • ¥15 Bug traq 数据包 大概什么价
  • ¥15 在anaconda上pytorch和paddle paddle下载报错
  • ¥25 自动填写QQ腾讯文档收集表