【Pytorch-gpu版安装教程,针对cuda11.1】

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

配了3天的环境终于true了,简单列举一下这三天遇到的问题和解决方法。
首先介绍一下成功的路径。
由于校园网不好,我使用了离线下载。
在下载torch之前,最好先打开命令行(输入cmd),在命令行中输入nvidia-smi查看显卡对应的版本:
【Pytorch-gpu版安装教程,针对cuda11.1】_第1张图片
然后打开这个网址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中首先要进入存放这两个包的地址中,然后新建虚拟地址
【Pytorch-gpu版安装教程,针对cuda11.1】_第2张图片
最后输入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版本,然后下载好对应的。

你可能感兴趣的:(pytorch,python,深度学习)