windows11安装GPU版本pytorch

Windows11 python3.9 安装gpu版本pytorch。
因为使用pytorch官网的安装指令最后一直FALSE，同时使用清华源也存在默认下载cpu版本的问题。
因此本文仅在Anaconda Prompt内手动输入指令进行安装，不需要去Pytorch官网复制命令。

安装CUDA CUDNN

安装CUDA可以参考这篇，本人的笔记本是GTX1060，安装了CUDA10.2

安装CUDNN可以参考这篇
后续安装pytorch时，cudnn只要开头的版本号对应就行，比如本人安装的cudnn版本是7.6.5，安装pytorch时，包名中有cudnn7_0即可

编辑系统变量、换源操作可以自行搜索

之前试了很多教程，包括还原后删除default
我认为直接指定版本下载的情况下，不需要删除default

主要步骤

1 首先执行以下代码清除缓存，一路y下去就行

conda clean --all

2 然后执行以下代码搜索

conda search pytorch

找到需要的对应的版本，将需要的包名、版本号、build连起来：name=version=build，比如本人安装的版本号如下所示

找到后按如下格式输入：

conda install pytorch=1.10.0=py3.9_cuda10.2_cudnn7_0

安装后可以输入指令查看版本

conda list

torchaudio与torchvision同理：

torchaudio:

conda search torchaudio

本人安装的torchaudio版本：

conda install torchaudio=0.10.0=py39_cu102

torchvision:

conda search torchvision

本人安装的版本，安装时按y就行

conda install torchvision=0.11.0=py39_cu102

本人的安装顺序是pytorch torchaudio torchvision
另外本人真的超级新手，如果有错误请指出

参考感谢

[1]: anaconda虚拟环境安装pytorch gpu版本
[2]: conda安装pytorch
[3]: win10及win11安装cuda10.2+cudnn+pytorch超详细过程