在VS Code中安装torch==1.9.1失败,求助解决版本匹配问题
解决torch==1.9.1安装失败问题
问题原因
torch 1.9.1的官方PyPI包已不再提供,直接通过pip install torch==1.9.1无法找到对应版本;同时该版本仅支持Python 3.6~3.9,Python 3.10及以上版本本身就不兼容这个版本的torch。
解决方案
方案1:通过Torch官方历史镜像安装
如果必须使用torch 1.9.1,可通过官方提供的历史版本安装链接指定源:
- CPU版本:
pip install torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html - CUDA 11.1版本(适配支持CUDA 11.1的环境):
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html
方案2:升级到兼容当前Python版本的Torch稳定版
放弃1.9.1版本,使用适配当前Python版本的最新稳定版,同时调整requirements.txt中的依赖版本:
修改后的requirements.txt内容:
torch>=2.4.0 torchvision>=0.19.0 scikit-learn==0.24.1 numpy==1.20.1 matplotlib==3.3.4 tqdm==4.59.0
执行安装命令:
pip install -r requirements.txt
注:若scikit-learn 0.24.1、numpy 1.20.1与新版torch出现依赖冲突,可适当升级这两个包的版本。
方案3:使用Python 3.8虚拟环境(仅针对必须用1.9.1的场景)
torch 1.9.1对Python 3.8的支持最完善,创建Python 3.8的虚拟环境后再尝试安装:
# 以conda创建虚拟环境为例 conda create -n torch191 python=3.8 conda activate torch191 # 执行方案1中的对应安装命令 pip install torch==1.9.1+cpu torchvision==0.10.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
内容的提问来源于stack exchange,提问作者polifolli
相关产品推荐
相关产品推荐

