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

关注

原创最新推荐文章于 2026-03-12 06:01:27 发布 4.6k 阅读 AI 写同款· GEO 优化 ›

从气象预报到物流规划：多尺度地理加权回归(MGWR)在ArcGIS Pro中的实战应用本文深入探讨了多尺度地理加权回归(MGWR)在ArcGIS Pro中的实战应用。通过对比传统GWR模型，MGWR的核心优势在于能识别不同解释变量的独特空间作用尺度，有效解决空间异质性问题。文章结合气象预报误差修正与物流网点选址两个案例，详细讲解了从数据准备、参数设置、带宽优化到结果解读的全流程，为地理数据分析师处理复杂空间数据提供了强大工具。阅读详情

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

当我们输入以下代码

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

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

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

这样就解决了以上的问题

收录于