【windows10下安装GPU版pytorch简明教程，仅需两步骤，以及解决报错：pytorch 无法导入torch加载..\caffe2_nvrtc.dll”或其依赖项之一时出错】

1.pytorch版本说明

pytorch可以分为两个版本，CPU版和GPU版，如果电脑上有GPU，尽量装GPU版本，运行速度比较快
目前网络上很多教程装pytorch前要装CUDA，cuDNN，以及配置环境虚拟变量，经过本人安装经历，这些都不需要。安装pytorch时会安装相应的包CUDA，cuDNN。

2.准备工作

2.1安装Anaconda（已经安装过就可以跳过）

1、Anaconda安装
下载anaconda，清华镜像下载或者官网安装，这一点可以参考安装anaconda或者其他人的说明

2.2查看cuda版本

在cmd命令行中输入nvidia-smi
nvidia-smi

可以看到右上角的CUDA Version是12.1，这代表可以满足的版本，只需要确认这个版本比装pytorch（官网给的时11.7,11.8）时大就行。

3.安装pytorch

3.1下载

pytorch的官网安装即可，下载pytorch
https://pytorch.org/get-started/locally/

然后选择Windows版本，package选择conda,comupte platform CUDA是GPU版本，CPU是CPU版本。
然后复制run this command的那行代码：conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia，

3.2打开Anaconda prompt

3.3运行复制的那段代码

enter 接下来会提示，选择[y/n].选择y然后直到安装结束，大概15分钟，也很快的。

4测试是否安装成功

4.1 anaconda prompt中测试

prompt中启动python

然后在prompt运行下面这段代码，能print出来，就安装成功了.逐行复制进去，然后enter，其实第2行能print出来就成功了。

import torch
print("Support CUDA ?: ", torch.cuda.is_available())
x = torch.tensor([10.0])
x = x.cuda()
print(x)
y = torch.randn(2, 3)
y = y.cuda()
print(y)
z = x + y
print(z)

CUDA是True。

4.2spyder中测试(重点来了）

这一个，我试了很多次，多种方法，最后发现其实很简单。在spyder中运行时报错：

OSError: [WinError 126] The specified module could not be found. Error loading "C:\Users\alan\anaconda3\lib\site-packages\torch\lib\caffe2_nvrtc.dll" or one of its dependencies.

先说结果参考网站
https://stackoverflow.com/questions/74552442/cannot-import-torch-error-loading-caffe2-nvrtc-dll-or-one-of-its-dependencie
其实很简单，因为很明显前面anaconda prompt中已经可以运行了，说明我们已经安装成功了，但是spyder为什么不能用呢，其实就是编译器的问题。
我们先从anaconda prompt中打开spyder就好了，然后再运行spyder就行了。


至此，pytorch GPU版本安装成功！！
那么有的人可能会说每次都从先从anaconda prompt中打开spyder太麻烦，其实只要打开一次，后面我们正常双击spyder的图标打开也能运行了。

附：关于4.2spyder中测试报错：OSError: [WinError 126] The specified module could not be found. Error loading “C:\Users\alan\anaconda3\lib\site-packages\torch\lib\caffe2_nvrtc.dll” or one of its dependencies.
我尝试的一些其他方法，但是对于我来说都不行，也供大家参考：
1、改变spyder中pythonpython解释器的路径
2、评论里有说从Anaconda运行的，但是不够清楚，只有一句话
3、下载pytorch时使用pip不用conda，但是我要卸载并下载，我没试过
4、也是使用pip不用conda，但是我要卸载并下载，我没试过