目录
1.背景:显卡驱动最高支持cu111,打算安装 pytorch1.10.0/cu111(torch1.10.0+cu111、 torchvision0.11.0+cu111)
2.报错:Error:no matching distribution found for torchvision==0.11.0+cu111
3.分析原因:torchvision0.11.0/cu111这个版本没有win环境下的安装包,所以显示torch1.10.0/cu111找不到对应的torchvision0.11.0/cu111
4.解决:改变思路,换安装 torch1.9.1/cu111 + torchvision0.10.1/cu111 (pytorch1.9.1/cu111),torch和torchvision版本匹配、且都有win下的安装包
1)电脑显卡环境:
(此显卡驱动最高支持cu111。于是,打算安装cuda11.1版本的pytorch)
2)在虚拟环境中,按照pytorch官网给的命令安装 pytorch1.10.0/cu111(torch1.10.0+cu111、 torchvision0.11.0+cu111):
# v1.10.0
# Wheel
# Linux and Windows
# CUDA 11.1
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
安装时报错,找不到对应的torchvision版本:
Error:no matching distribution found for torchvision==0.11.0+cu111
首先分析一下原因。
在https://download.pytorch.org/whl/cu111/torch_stable.html网址里,可以看到cu111下的torch、torchvision的目前可以用的所有版本。
1)可以看到,对于torch,有win环境下torch1.10.0/cu111这个版本的安装文件 torch-1.10.0+cu111-cp38-cp38-win_amd64.wh:
2)但是,对于torchvision,torchvision0.11.0/cu111这个版本是没有win环境下的安装包的,全都是linux环境下的安装包。。。
所以,会出现上面的报错,说torch1.10.0/cu111找不到对应的torchvision0.11.0/cu111。。。。
总而言之,也就是说,在win环境下安装pytorch1.10.0/cu111(torch1.10.0+cu111、 torchvision0.11.0+cu111)是不可能安装成功的。。。。因为它就没有win环境下对应的torchvision安装包呀,你怎么可能安装成功。。。肯定会报错的。。。。。
找好torch和torchvision对应好的安装包,再安装,就能解决了。
于是找cu111下,torchvision在win环境下的安装包。
1)从下图可以看到,torchvision0.11.0、0.11.1、0.11.2、0.11.3的安装包全都是linux环境下的。win环境下的torchvision安装包最高是0.10.1,那就安装这个吧!
2)决定安装torchvision0.10.1/cu111。
在命令行中:pip install torchvision-0.10.1+cu111-cp38-cp38-win_amd64.whl。从下图可以看到它会自动下载torch1.9.1,因为根据torch和torchvison匹配表 torchvision0.10.1和torch1.9.1是匹配的。
3)所以,这就相当于是安装 torch1.9.1/cu111 + torchvision0.10.1/cu111 (pytorch1.9.1/cu111),torch和torchvision版本匹配、且都有win下的安装包。那么,最终的解决方案就是:使用下面的命令安装torch1.9.1/cu111 + torchvision0.10.1/cu111 (pytorch1.9.1/cu111),这次肯定不会报错了,完美!
pytorch官网上的安装命令:
# CUDA 11.1
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
详细安装步骤可参考下一篇博客:PyTorch1.9.1 GPU版本安装(python3.8+pyTorch1.9.1, torch1.9.1/cu111 + torchvision0.10.1/cu111)
参考:
安装pytorch报错:no matching distribution found for torchvision =0.11.0+cu111_123GJ123的博客-CSDN博客
Pytorch1.10安装记录(CUDA11.3)_太阳花的小绿豆的博客-CSDN博客