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资源是非常有用的。

TensorFlow详细配置(Python版本)

评论已关闭

推荐阅读