介绍
如果您和我一样,就会知道在新设备上设置 TensorFlow 可能会很痛苦。在本教程中,我将介绍在 Windows 11 上安装 CUDA、cuDNN 和 TensorFlow。本文的目标是使该过程尽可能轻松。本文假设您的 PC 上已安装了兼容 CUDA 的 GPU。
安装步骤
按顺序执行每个步骤非常重要。如果您不按顺序执行这些步骤,那么您将不得不重新开始该过程,而没有人愿意这样做。所以,事不宜迟,让我们直接进入正题。
我们需要安装的软件:
Microsoft Visual Studio(2019社区版)
NVIDIA CUDA 工具包
NVIDIA cuDNN
Python(我个人使用anaconda)
TensorFlow
第 1 步:删除所有当前 NVIDIA 驱动程序
转到 Windows 上的搜索栏并搜索“添加或删除程序”。您现在应该看到已安装应用程序的列表。向下滚动,直到看到任何显示 NVIDIA 的内容。它应该类似于下图。