tensorflow配置是否成功怎么查看-开发者知识库平台

tensorflow配置是否成功怎么查看

tensorflow

840

2024/4/11 13:52:35

栏目: 深度学习

要查看TensorFlow的配置是否成功，可以在Python代码中使用以下方法：

import tensorflow as tf

print("TensorFlow version: ", tf.__version__)
print("Is TensorFlow using GPU: ", tf.test.is_built_with_cuda())
print("Is TensorFlow using GPU support: ", tf.test.is_built_with_gpu_support())

这段代码会输出TensorFlow的版本号以及是否成功配置了GPU支持。如果输出结果显示TensorFlow正在使用GPU，则说明配置成功。

辰迅云「云服务器」，即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘，价格低至29元/月。点击查看>>

tensorflow配置是否成功怎么查看

最新知识库

相关标签