tensorflow 1.13 (tf 1.13)中keras设置GPU显存大小

最新推荐文章于 2023-02-28 19:55:09 发布

醉意流年go

最新推荐文章于 2023-02-28 19:55:09 发布

阅读量362

点赞数

分类专栏： tensorflow 文章标签： tensorflow keras python

本文链接：https://blog.csdn.net/u010626747/article/details/125082153

版权

tensorflow 专栏收录该内容

25 篇文章 0 订阅

订阅专栏

import os
import numpy as np
import sys
import tensorflow as tf
from tensorflow.python.keras.backend import set_session

# tf配置
path = os.path.abspath(os.path.dirname(os.path.dirname(__file__)))
os.environ["CUDA_VISIBLE_DEVICES"] = "0"
config = tf.ConfigProto(allow_soft_placement=True)

config.gpu_options.per_process_gpu_memory_fraction = 0.1 # 占用GPU90%的显

sess = tf.Session(config=config)
graph = tf.get_default_graph()

# IMPORTANT: models have to be loaded AFTER SETTING THE SESSION for keras!
# Otherwise, their weights will be unavailable in the threads after the session there has been set
set_session(sess)