导入以下代码控制GPU内存占用，爆出与tensorflow2.0版本不兼容问题。

只需将下面的代码：

import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"
config = tf.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))

改为：

import os
import tensorflow as tf
os.environ["CUDA_VISIBLE_DEVICES"] = "2"`
config=tf.compat.v1.ConfigProto()
config.gpu_options.per_process_gpu_memory_fraction = 0.3`
tf.compat.v1.keras.backend.set_session(tf.compat.v1.Session(config=config))

`

即可。

# tensorflow
Logo
AI编程社区

汇聚全球AI编程工具，助力开发者即刻编程。

更多推荐

第一篇：AI编程工具发展史-从工具到伙伴的蜕变

AI编程工具如何3年改变开发方式：从GitHub Copilot的震撼到日常生产力革命 摘要：GitHub Copilot的问世彻底改变了编程方式，使AI编程助手从简单补全进化为真正的"编程伙伴"。数据显示76%开发者已使用AI工具，效率提升35-50%。文章回顾了编程工具四个发展阶段：石器时代的痛苦摸索（1990s-2010s）、青铜时代的机器学习曙光（2010s-2018）

avatar AI编程社区

开发AI编程工具的方案分析

大语言模型的 Token 成本是项目运营的核心成本之一，需要实现精确的上下文控制、智能压缩策略、增量更新机制，在保证功能完整性的前提下最小化 Token 消耗，这涉及复杂的文本压缩算法、上下文窗口管理、模型调用优化等技术。：必须与 VS Code、IntelliJ、Vim 等主流开发环境实现无缝集成，涉及插件开发、协议适配、实时同步等复杂技术。：不仅要理解代码语法，更要理解代码的业务逻辑、设计模式

avatar AI编程社区

Copilot移动开发指南：Android_iOS双平台实战

先写一个User// 用Retrofit定义GitHub API接口，包含获取用户列表的GET请求（路径：/users）suspend fun getUsers(): Response<List<User>> // 用suspend函数支持Coroutines先写Todocase avatarUrl = "avatar_url" // 映射JSON的蛇形命名在Android Studio和Xcod

avatar AI编程社区