如何判断TensorFlow是否使用GPU？

• 日期: 2028-10-14 10:30:17
• 标签: 常规

在 TensorFlow 中，可以通过以下代码来判断是否使用了 GPU：

import tensorflow as tf

tf.config.list_physical_devices('GPU')

如果输出为空列表，表示没有使用 GPU。如果输出一个非空列表，表示使用了 GPU。

原文地址: https://www.cveoy.top/t/topic/pXyF 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录