TensorFlow 2.0之后动态分配显存方式
作者:scluis 时间:2023-12-20 05:25:17
TensorFlow 2.0之后动态分配显存
import tensorflow as tf
config = tf.compat.v1.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.compat.v1.Session(config = config)
tf.compat.v1.keras.backend.set_session(sess)
keras和tensorflow动态分配显存
tensoflow或者基于tensorflow的keras,好像默认会直接将GPU中所有的显存都读过来,然后慢慢用。
这样如果是服务器是的话,别人就没法用了,这里转载了动态分配显存的方法:
tensorflow下,加入代码:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True #设置动态分配显存
session = tf.Session(config=config, ...)
或者:
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3 #只一次调用30%的显存
session = tf.Session(config=config, ...)
对于keras,如果使用tensorflow作为后端,多加一句话就好了:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
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)) # 此处不同
来源:https://blog.csdn.net/weixin_42619772/article/details/122547421
标签:TensorFlow,动态分配,显存
0
投稿
猜你喜欢
按钮的反馈
2009-01-01 20:06:00
Python调用VBA实现保留原始样式的表格合并方法
2022-10-18 02:40:45
pytorch中的model=model.to(device)使用说明
2023-02-23 15:07:48
不用script仅用css编写无限级弹出菜单
2008-04-24 14:03:00
CSS高级文字排版的实例
2009-03-24 20:56:00
asp fso删除非空目录的例子
2008-11-26 20:03:00
python使用正则表达式替换匹配成功的组
2021-12-13 07:22:03
Python函数式编程
2023-12-29 10:58:46
利用python Selenium实现自动登陆京东签到领金币功能
2021-11-09 12:00:33
python中Tkinter复选框Checkbutton是否被选中判断
2023-10-20 16:41:19
Python pip更新的两种方式详解
2022-07-06 20:00:29
python中的txt文件转换为XML
2021-12-05 10:45:48
python导入csv文件出现SyntaxError问题分析
2023-12-12 04:29:57
Django中QuerySet查询优化之prefetch_related详解
2022-02-03 04:18:44
python中数字列表转化为数字字符串的实例代码
2021-04-30 02:46:45
pandas 空的dataframe 插入列名的示例
2023-10-27 07:48:18
python实现手机销售管理系统
2023-01-16 00:02:04
python制作简单五子棋游戏
2021-09-13 06:49:47
详解python中各种文件打开模式
2023-08-30 09:10:30
sql server not in 语句使程充崩溃
2012-01-05 19:05:00