标签:compat memory bytes gpu v1 MEMORY tf import options
显存充足,但是却出现CUDA error:out of memory错误
添加:
import os
# 设置程序使用的GPU编号:我用6号和7号显卡
os.environ['CUDA_VISIBLE_DEVICES'] = '0,1'
import numpy as np
import keras
import keras.backend as K
import tensorflow.compat.v1 as tf
tf.disable_v2_behavior()
import keras.backend as K
gpu_options = tf.compat.v1.GPUOptions(per_process_gpu_memory_fraction=0.99)
gpu_options.allow_growth = True
sess = tf.compat.v1.Session(config=tf.compat.v1.ConfigProto(gpu_options=gpu_options))
标签:compat,memory,bytes,gpu,v1,MEMORY,tf,import,options 来源: https://blog.csdn.net/hanshuning/article/details/123215958
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。