【Windows】安装pytorch+cuda遇到的一些坑以及最简单直接的解决方法

【Windows】安装pytorch+cuda遇到的一些坑以及最简单直接的解决方法

折腾了一晚上没有搞好，今早搞好了，我总结一下我的过程，很简单

从官网找到下载链接，然后用迅雷下载下来，可能会很慢，迅雷还有免费加速，或者充个会员

1. 资源下载网址： ，https://download.pytorch.org/whl/torch_stable.html；
2. 找到自己还要下载的版本，需要查看自己的显卡当前的cuda版本，nvdia控制面板，左下角系统信息，上面第二个选项卡组件可以看到自己的cuda版本（第三行），比如我这里是CUDA11.5 3.然后去官网下载对应版本的，举个例子：（我也不知道为什么我舍友的电脑是115版本，我之前安装的那台电脑是111版本，咱们以111版本举例）我需要的版本是，python3.9，加pytorch1.8.1+cuda11.1于是找到了这个点击他就会开始下载，然后右键它复制下载链接，在迅雷新建任务，用迅雷下载，如果用迅雷会员加速的话，2小时差不多就可以加载完成，需要的其他torchvision和torchaudio以相同的方法下载即可。

安装本地的whl包

进入对应版本的python版本的目录一般是：C:\Users\用户名\AppData\Local\Programs\Python\Python39\ Scripts在这个Scripts中以管理员身份打开cmd，之后用pip的安装指令pip install .whl（注：.whl就是你下载的那个文件的目录，例如：“pip install D:\Desktop\torchvision-0.9.1+cu111-cp39-cp39-win_amd64.whl”）然后等几秒钟就会告诉你安装成功了，其他的包也是这个方法

最后使用包

如果是和我一样是pycharm的话，在settings中找到projects中编译器inter啥啥啥的没记住名字，然后选择编译器的话就要用刚刚安装的那个python（看后面的路径和刚刚安装的路径一样的那个python）
如果没有添加的话，就要添加到编译器中，就按照这个流程自左向右操作，把刚刚配置的那个python添加进去

然后在这里的软件包找到torch然后就可以看到版本是torch+cu111就是安装成功了

最后查看一下是否成功，在编译器中输入

import torch
print(torch.cuda.is_availble())

如果输出

True

那么就安装成功啦！