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下载地址
我选择的上述红色框出来的这个,这里一定要值得注意下: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/
一般来说,你先确定好你想要安装的torch版本,然后再根据官网上推荐的torchvision版本安装即可~