tf.test.is_built_with_cuda() 应该输出什么？

Question

我通过

tensorflow-gpu

安装了

conda install tensorflow-gpu

。

但是，我安装的计算机没有GPU。 我这样做是因为我正在将 conda env 移植到离线计算机上。

当我

import tensorflow as tf

并致电

tf.test.is_built_with_cuda()

时，结果是

False

。 尽管计算机没有 GPU，但我期望结果是

True

，因为我安装了 led张量流GPU。 在这种情况下 tf.test.is_built_with_cuda() 的输出应该是什么？

为什么我会得到

False

？

Answer 1

我刚刚使用之前运行 CUDA GPU 的服务器进行了测试，但后来删除了 GPU - 与您的情况非常相似。我已经安装了

tensorflow-gpu==2.5.0

。

>>> import tensorflow as tf
2021-07-28 18:42:44.236258: I tensorflow/stream_executor/platform/default/dso_loader.cc:53] Successfully opened dynamic library libcudart.so.11.0
>>> tf.test.is_built_with_cuda()
True

这里记录了类似的问题。因此，你很可能缺乏 CUDA/CUDNN。

Answer 2

就我而言，此问题指令返回 true。

请尝试按照测试的构建配置中所述安装带有 CUDA 11.0 和 cuDNN 8 的 TensorFlow v2.4，并检查您是否遇到相同的问题，请参阅此问题返回 true，