Pytorch-gpu版安装教程，针对cuda11.1

Posted 2023-01-04 iiidep

针对cuda11.1版本安装pytorch-gpu版本@TOC

配了3天的环境终于true了，简单列举一下这三天遇到的问题和解决方法。
首先介绍一下成功的路径。
由于校园网不好，我使用了离线下载。
在下载torch之前，最好先打开命令行（输入cmd）,在命令行中输入nvidia-smi查看显卡对应的版本：

然后打开这个网址https://download.pytorch.org/whl/torch_stable.html，寻找版本对应的torch和torchvision进行下载。
此处介绍gpu版本，因为有英伟达显卡，所以选择下载gpu版本，此处我下载的是这两个
cu111/torch-1.9.0%2Bcu111-cp38-cp38-win_amd64.whl
cu111/torchvision-0.10.0%2Bcu111-cp38-cp38-win_amd64.whl
（cu111是对应cuda11.1），选择windows版本而不是linux，因为这里我有遇到过，先下载了linux版本然后报错，因为文件里只能识别win_amd64类似的名字。所以要选对版本。
我是将它们下载到了D盘的安装包里，然后在anaconda prompt中首先要进入存放这两个包的地址中，然后新建虚拟地址

最后输入python,import torch若未报错，则安装成功，输入torch.cuda.is_available()若显示true,则说明下载版本与本机显卡适配。
遇到的问题：
1.之前有碰到说conda安装是在D盘，pip是在C盘的说法，所以一开始使用conda安装，但是报错Solving environment: failed with initial frozen solve.搜索这个错误又说是anaconda版本有问题，然后我改成pip，直接安装成功了。
2.建议网不好的都用离线安装，因为使用镜像又会出现其他很多的问题，无力列举了。。不行就切手机热点
3.强烈建议先看看自己cuda版本，然后下载好对应的。