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中运行时报错:
先说结果参考网站
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,但是我要卸载并下载,我没试过