Linux 拨号vps windows公众号手机端

tensorflow配置是否成功怎么查看

lewis 5年前 (2020-02-10) 阅读数 9 #大数据
文章标签 tensorflow

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

版权声明

本文仅代表作者观点,不代表米安网络立场。

发表评论:

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

热门