tensorflow配置是否成功怎么查看

lewis 2020-02-10 18次阅读

要查看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,则说明配置成功。



发表评论:

◎欢迎参与讨论,请在这里发表您的看法、交流您的观点。