PyTorch简易安装方法(100%成功)

PyTorch简易安装方法(100%成功)

一般我们习惯用pip install xxx,同时使用清华源用来加速,这种方法安装99%的库都没啥大问题,但是有时候安装pytorch的时候就会发现问题。笔者反正每次用该方法安装torch总会遇到刚开始下载速度很快,但是下载到快一半的时候下载速度就会极具下降,所以笔者推荐一个简易安装的方法,就是直接用wheel文件来安装。


笔者用下载torch==1.11.0,torchvision==0.12.0为例子:

1.下载对应版本的torch wheel文件

下载地址:https://download.pytorch.org/whl/torch/

根据上述找到对应版本的torch下载地址

PyTorch简易安装方法(100%成功)_第1张图片
我选择的上述红色框出来的这个,这里一定要值得注意下:1.torch版本 2.cuda版本或者cpu 3.python版本

2.pip安装torch

pip install your_wheel_path

此时,可能会下载依赖库,一般来说,依赖库下载就比较快。

3.torchvision安装

torchvision安装同理torch的安装,它的地址为:

torchvision下载地址:https://download.pytorch.org/whl/torchvision/

值得注意的是,torch与torchvision版本要进行对应,具体的可查看:

版本对应查看地址:https://pytorch.org/get-started/previous-versions/

PyTorch简易安装方法(100%成功)_第2张图片
一般来说,你先确定好你想要安装的torch版本,然后再根据官网上推荐的torchvision版本安装即可~

你可能感兴趣的:(pytorch,杂文,目标检测--小细节问题,pytorch,深度学习,算法,计算机视觉,人工智能)