如何解决tf.test.is_built_with_cuda() 应该输出什么?
我通过 tensorflow-gpu
安装了 conda install tensorflow-gpu
。
但是,我安装的计算机没有 GPU。我做了这个 b/c 我正在将 conda env 移植到离线计算机。
当我 import tensorflow as tf
并调用 tf.test.is_built_with_cuda()
时,结果是 False
。即使计算机没有 GPU,我也期望结果是 True
,因为我安装了
led tensorflow-gpu。
在这种情况下,tf.test.is_built_with_cuda() 的输出应该是什么?
为什么我会得到 False
?
解决方法
我刚刚使用之前运行 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
记录了类似的问题 here。因此,您很可能缺少 CUDA/CUDNN。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。