解决tensorflow2.0版本的gpu内存占用问题(module ‘tensorflow‘ has no attribute ‘ConfigProto‘解决办法)

时间:2025-04-18 22:21:43

这个文章当作我学习tensorflow的一个笔记

当我们输入以下代码

import tensorflow as tf
config = ()
config.gpu_options.visible_device_list = '0'
config.gpu_options.allow_growth = True
sess = (config=config)

可能会出来
module ‘tensorflow’ has no attribute ‘ConfigProto’
原因之一是这个代码适用于tensorflow1.0版本,而电脑上安装的是2.0的版本
因此应该输入

import tensorflow as tf 
import os
['CUDA_VISIBLE_DEVICES']="0"
config=.() 
config.gpu_options.visible_device_list = '0' 
config.gpu_options.allow_growth = True 
sess=.(config=config)


这样就解决了以上的问题