python - 我可以在 TensorFlow 上同时运行 AMD GPU 和 NVIDIA GPU 吗？

Question

简单的问题：我可以在 TensorFlow 中同时运行双 GPU 设置（如下所示）吗？

 1 AMD RX 480 and 1 NVIDIA 3070
 (ROCm 3.5.1 and CUDA 11)

我试图通过安装 TensorFlow (2.4) 和 TensorFlow ROCm(2.4) 以及所有适当的驱动程序来做到这一点。不幸的是，TensorFlow 只选择其中之一，这取决于是否安装了 TensorFlow ROCm。

我怎样才能解决这个问题？

如果不可能，原因是什么？

标签： pythontensorflowdeep-learning

Answer 1

我真的不认为你能做到。Tensorflow-rocm 是 Tensorflow 的一个分支。AMD 已经做出了很大改变，以实现与 GPU 的兼容性。因此，由于它们是不同的包，它们不能完全运行。