【无标题】

最新推荐文章于 2025-06-24 17:22:17 发布

Coffee_Monster_boy

最新推荐文章于 2025-06-24 17:22:17 发布

阅读量77

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习入门文章标签： tensorflow 深度学习人工智能

本文链接：https://blog.youkuaiyun.com/jason_xiaoyi/article/details/124767691

深度学习入门专栏收录该内容

2 篇文章

订阅专栏

这段代码展示了如何在TensorFlow中动态管理GPU内存。通过`tf.config.experimental.list_physical_devices('GPU')`获取GPU列表，并使用`tf.config.experimental.set_memory_growth`启用GPU内存按需增长，从而有效利用GPU资源。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# 代码中插入：
import tensorflow as tf
gpus = tf.config.experimental.list_physical_devices('GPU')
for gpu in gpus:
    tf.config.experimental.set_memory_growth(gpu, True)

# 问题解决！