TensorFlow 2.0之后动态分配显存方式
作者：scluis 发布时间：2023-12-20 05:25:17
标签：TensorFlow,动态分配,显存
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
0
投稿猜你喜欢
前言：python利用matplotlib库中的plt.ion()函数实现即时数据动态显示：1.非定长的时间轴代码示例：# -*- codi- 从codered到nimda等，一大堆蠕虫把原来需要人工利用的漏洞都变成了程序自动利用了，大家还想去手工操作这些IIS漏洞么?让我们调整重心
- 经过摸索和实践，我把自己的解决方法，写在下面： 说明： 我的Oracle客户端的版本是 oracle 9i， 安装client端的时候，不能
- 本文实例为大家分享了python实现图片识别汽车的具体代码，供大家参考，具体内容如下准备工作1、登陆开发者控制台2、安装 pip insta
- 代码如下:Create trigger tri_wk_CSVHead_History on wk_CSVHead_History --声明
- 本篇文章所编写的代码，已经放到了gitee上了:gitee.com/pdudo/golea…本文所依赖的环境为:在目前
- 本文实例讲述了Python实现PS图像调整黑白效果。分享给大家供大家参考，具体如下：这里用Python 实现 PS 里的图像调整–黑白，PS
- MySQL的常见操作在这里先做一下总结，已经整合到代码里面，经过检验无误。/*创建一个数据库*/create database xuning
- 无意间碰到的一个大神整理的Python学习思维导图，感觉对初学者理清学习思路大有裨益，非常感谢他的分享。14 张思维导图基础知识数据类型序列
- scrapy框架概述：Scrapy，Python开发的一个快速,高层次的屏幕抓取和web抓取框架，用于抓取web站点并从页面中提取结构化的数
- 【简介】django-admin.py是Django的一个用于管理任务的命令行工具。本文将描述它的大概用法。另外，在每一个Django pr
- 因为写别的程序想要一边遍历一边删除列表里的元素，就写了一个这样的程序进行测试，这样写出来感觉还挺简洁的，就发出来分享一下。代码l=list(
- 这篇文章主要介绍了基于python实现从尾到头打印链表,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋
- python爬虫模块PyQuery简介PyQuery库也是一个非常强大又灵活的网页解析库，如果你有前端开发经验的，都应该接触过jQuery,
- PIL（Python Imaging Library）是Python中一个强大的图像处理库，但目前其只支持到Python2.7pillow是
- MySQL中可以使用rename table这个SQL语句来修改表名。rename table这个SQL语句来修改表名的基本语法是：RENA
- Python中多线程使用到Threading模块。Threading模块中用到的主要的类是Thread，我们先来写一个简单的多线程代码：#
- 本文实例为大家分享了python实现简单俄罗斯方块游戏的具体代码，供大家参考，具体内容如下import pygame,sys,random,
- UI自动化测试过程中，可能会出现因测试环境不稳定、网络慢等情况，如果不做任何处理的话，会出现无法定位到特定元素而报错，导致自动化测试无法顺利
- 这篇文章主要介绍了Python 实现两个矩形重合面积代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需
