如果您最近购买了一台安装了 Windows 11 的新笔记本电脑,并且有兴趣使用 PyTorch 进行一些深度学习,那么您来对地方了。
在本文中,我将分步指导如何安装支持 GPU 的 PyTorch。
我的笔记本电脑是带有 RTX 3050 显卡的 HP Omen 16。
先决条件之一是您需要在系统中安装“ conda ”。你在使用“ Anaconda”安装 python 时得到这个。请检查链接以安装“ Anaconda ”
步骤
首先,您需要在系统上安装 Cuda 工具包,否则每当您尝试在Powershell中运行“ nvcc ”命令时都会出现错误。
网友评论