我使用的是tensorflow 2.3版。当检查可用设备时，它显示xla-gpu。我的代码只能在gpu上运行（不能在xla-gpu上运行）。任何人都可以告诉我为什么它会这样显示以及解决方案。
7kqas0il1#
当从bazel中的源构建tensorflow时，有一个选项可以设置XLA启用或禁用。如果您已经从源构建，请确保您在XLA选项中选择了“否”。如果使用pip安装，那么您可以尝试其他版本的tensorflow，如2.4。它应该工作
1条答案
按热度按时间7kqas0il1#
当从bazel中的源构建tensorflow时，有一个选项可以设置XLA启用或禁用。如果您已经从源构建，请确保您在XLA选项中选择了“否”。
如果使用pip安装，那么您可以尝试其他版本的tensorflow，如2.4。它应该工作