如果您最近购买了一台安装了 Windows 11 的新笔记本电脑，并且有兴趣使用 PyTorch 进行一些深度学习，那么您来对地方了。

在本文中，我将分步指导如何安装支持 GPU 的 PyTorch。

我的笔记本电脑是带有 RTX 3050 显卡的 HP Omen 16。

先决条件之一是您需要在系统中安装“ conda ”。你在使用“ Anaconda”安装 python 时得到这个。请检查链接以安装“ Anaconda ”

步骤

首先，您需要在系统上安装 Cuda 工具包，否则每当您尝试在Powershell中运行“ nvcc ”命令时都会出现错误。