在 Windows 11 PC 上安装支持 CUDA、cudNN 和 GPU 的最新 TensorFlow 版本

介绍

如果您和我一样,就会知道在新设备上设置 TensorFlow 可能会很痛苦。在本教程中,我将介绍在 Windows 11 上安装 CUDA、cuDNN 和 TensorFlow。本文的目标是使该过程尽可能轻松。本文假设您的 PC 上已安装了兼容 CUDA 的 GPU。

安装步骤

按顺序执行每个步骤非常重要。如果您不按顺序执行这些步骤,那么您将不得不重新开始该过程,而没有人愿意这样做。所以,事不宜迟,让我们直接进入正题。

我们需要安装的软件:

Microsoft Visual Studio(2019社区版)
NVIDIA CUDA 工具包
NVIDIA cuDNN
Python(我个人使用anaconda)
TensorFlow

第 1 步:删除所有当前 NVIDIA 驱动程序

转到 Windows 上的搜索栏并搜索“添加或删除程序”。您现在应该看到已安装应用程序的列表。向下滚动,直到看到任何显示 NVIDIA 的内容。它应该类似于下图。

在这里插入图片描述
现在检查并删除上面标有 NVIDIA 的所有内容。不要担心删除它们,我们将在后面的步骤中将它们安装回来。

步骤 2:安装 Visual Studio(不是 VS code)

安装 CUDA 工具包需要 Visual Studio。如果您尝试在未安

猜你喜欢

转载自blog.csdn.net/iCloudEnd/article/details/132423014