在 VSC 中使用 GPU 进行 TensorFlow

Question

我刚刚将 Windows 升级到第 11 版，并努力强制 TensorFlow 在 Visual Studio 代码中使用 GPU。

所以我有几个问题：

1. 是否可以在 Windows 11 上实现，或者我应该将其降级到版本 10？
2. 为此，使用从网站下载的 Anaconda 的 python 或 Py 更好吗？
3. 是否可以使用 VSC 制作它，还是应该下载 Visual Studio？

我尝试了所有软件，除了所有软件的正确兼容版本（CUDA、Nvidia 驱动程序、cuDNN：我下载了它们的最新版本）。 我有 GeForce RTX3050 笔记本电脑，Win11。

Answer 1

回答你的第一个问题，这取决于 TF 2.10 之后的 Tensorflow 版本。

TensorFlow 2.10 是最后一个支持 GPU 的 TensorFlow 版本 本机 Windows。从 TensorFlow 2.11 开始，您将需要 在WSL2中安装TensorFlow，或者安装tensorflow或tensorflow-cpu 并且可以选择尝试 TensorFlow-DirectML-Plugin

这是参考。

其次，我建议使用单独的虚拟环境并使用 pip 来解决此问题。 Anaconda 可能会带来额外的并发症。

关于GPU的使用，你可以参考我类似回答的问题这里。