TensorFlow详细配置(Python版本)
# 导入TensorFlow库
import tensorflow as tf
# 获取TensorFlow版本
tf_version = tf.__version__
# 获取GPU支持情况
gpu_available = tf.test.is_gpu_available()
# 打印详细配置信息
print("TensorFlow Version:", tf_version)
print("Is GPU available:", gpu_available)
# 如果GPU可用，打印CUDA和cuDNN的版本
if gpu_available:
device_name = tf.test.gpu_device_name()
print("GPU device name:", device_name)
print("CUDA version:", tf.config.list_physical_devices('GPU'))
print("cuDNN version:", tf.config.list_physical_devices('GPU'))这段代码可以帮助你检查TensorFlow的版本，检查GPU是否可用，以及获取GPU的设备名称和相关的CUDA和cuDNN版本信息。这些信息对于确保TensorFlow正确地使用GPU资源是非常有用的。
评论已关闭