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
投稿猜你喜欢
接管Dingo-api的错误如上图所示，AppServiceProvider.php中的register()方法中添加如下代码\API::e- 页面重构需要考虑的一个重点是XHTML代码语义化，就算是在无任何CSS样式修饰的情况下也能给他人在阅读时带来便利，甚至可以夸张点说在搜索引擎
- 主要实现功能1、用户输入用户名，在用户名文件中查找对应的用户，若无对应用户名则打印输入错误2、用户名输入正确后，进行密码匹配。输入密码正确则
- 引言本文想要解决的问题是当DataFrame中某一列元素为不定长度的数组时，该如何对它们进行拆分分解为后续元素，从而进行进一步的提取操作，数
- 用PHP的json_encode来处理中文的时候, 中文都会被编码, 变成不可读的, 类似”\u***”的格式，如果想汉字不进行转码，这里提
- 本文实例讲述了Python实现基于KNN算法的笔迹识别功能。分享给大家供大家参考，具体如下：需要用到：Numpy库Pandas库手写识别数据
- 1.双击setup.exe.(出现安装向导界面) 2.在安装向导界面:选"基本安装",并选好主目录位置；创建启动数据库(
- 通信信息包是发送至MySQL服务器的单个SQL语句，或发送至客户端的单一行。在MySQL 5.1服务器和客户端之间最大能发送的可能信息包为1
- 本文实例讲述了Python中列表与元组的乘法操作。分享给大家供大家参考，具体如下：直接上code吧，还可以这么玩儿列表乘法：li=[1,]l
- 1.什么是局部视图局部视图是在其他视图中呈现的视图。通过执行局部视图生成的HTML输出呈现在调用视图中。与视图一样，局部视图使用 .csht
- 在 MySQL 下，在进行中文模糊检索时，经常会返回一些与之不相关的记录，如查找 "％a％" 时，返回的可能有中文字符，
- 本文实例讲述了js对象基础用法。分享给大家供大家参考。具体分析如下：js对象在本质上与数组相同，都是存放一组数据。但创建方法有所不同，对象需
- FSO中除了可以对驱动器、文件夹的操作以外，功能最强大的就是对文件的操作了。它可以用来记数、内容管理、搜索还可生成动态HTML页面等等。一、
- 零、本讲学习目标了解面向对象编程思想掌握类和对象的定义和使用了解Python中的对象一、面向对象（一）程序员“面向对象”在现实世界中存在各种
- python中eval函数的用法十分的灵活，这里主要介绍一下它的原理和一些使用的场合。下面是从python的官方文档中的解释： &
- 说明：通过随机产生密码，然后将密码EMail给注册用户，你可以确认用户的EMail填写是否正确。自动产生的密码往往安全性更高，同时，你可以过
- 组合模式Composite Pattern是什么组合模式是一种结构型模式，它允许你将对象组合成树形结构来表示“部分-整体
- 有关pygal的介绍和安装，大家可以参阅《pip和pygal的安装实例教程》，然后利用pygal实现画世界地图。代码如下：#coding=u
- 一、爬山法简介爬山法（climbing method）是一种优化算法，其一般从一个随机的解开始，然后逐步找到一个最优解（局部最优）。 假定所
- 数据库是什么 在学习ACCESS之前，我们先了解一下什么是“数据库”。我们举个例子来说明这个问题：每个人都有很多亲戚和朋友，为了保持与他们的
