目录
该报错是在按官网方法用指令:
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
安装pytorch时出现的,以下是分析:
一、猜测
这个错误提示表明在指令提供的下载网址上没有找到符合要求的torchvision软件包版本,需要安装符合要求的版本。问题可能出在指定的版本号(0.11.2+cu111),这个版本可能不兼容系统或者Python环境又或者根本就不存在,可以尝试安装其他版本的torchvision软件包。
另外,可以查看Python环境是否与要求的CUDA版本兼容,以及查看你的系统是否安装了相应的CUDA驱动程序。你还可以尝试升级你的Python环境或者使用虚拟环境来解决这个问题。
二、验证
在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对应好的都在契合win的系统架构的安装包,再执行相关的安装指令就能解决了。
我的cuda是11.1的版本,所以用下面这个指令,因版本而异,大家自行匹配。使用下面的命令安装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.htm
四、检验
用以上方案安装后
进行验证:
没问题!