tensorflow - Tensorflow1.x 和 Tensorflow2.x 的 CUDA？

Question

在过去的几年里，我一直在使用带有 Tensorflow1 的 Sonnet，但是随着 Tensorflow2.0 的引入，我转向了那个系统。现在，我正在调试一些东西，我想回到 TF1 设置。自从我迁移到 TF2.0 后，我不再可以访问为 TF1 设置了 CUDA 的机器。所以我的问题是：

是否可以在 Windows 10 机器上进行 CUDA 设置，允许在 Tensorflow1.x 和 Tensorflow2.x 环境中工作？

Answer 1

每个 Tensorflow-gpu 版本都需要适当的 CUDA 和 cuDNN 版本。Tensorflow-gpu==1.15 和 Tensorflow-gpu==2.0 支持 CUDA 10.0 和 cuDNN 7.4。