任务管理器->性能
二、查看电脑可以安装的cuda版本Cuda版本根据电脑显卡驱动选择
查看方法有两种
1)打开NVIDIA控制面板->帮助->系统信息->组件
2)使用命令查看
注意:这里的cuda版本是可以安装的最高版本
三、安装pytorch进入PyTorch 官网
在如下界面选择合适的版本,推荐使用Stable(以及选择版本1.1以上的才有tensorboard)
如果没有找到自己所需的版本,点击previous versions,有更多选择
可以看到,我的GPU型号为RTX3050,最高可以安装的CUDA版本为11.2,加上有些版本装了很容易出问题,所以我参照了下面两篇博客选定了合适的版本
RTX3050安装CUDA cuDNN及pytorch记录_Gui SY的博客-CSDN博客
(Windows 11+RTX3060安装CUDA、cuDNN与Pytorch的正确版本_乔卿的博客-CSDN博客
最后进入你的python环境粘贴Pytorch官网提供的相应安装指令即可



