如何在虚拟环境中的 Windows 上使用 pip 和 python 3.8.3 安装 torch 和 torchvision？

Question

我为我的项目创建了一个虚拟环境，我正在尝试通过

torch

安装

pip

。

pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html

这给了我错误-

ERROR: Could not find a version that satisfies the requirement torch===1.5.1 (from versions: 0.1.2, 0.1.2.post1, 0.1.2.post2)
ERROR: No matching distribution found for torch===1.5.1

有些解决方案推荐使用conda，有必要吗？我可不可以只通过

pip

Answer 1

您可以尝试直接从wheels文件安装：

pip install https://download.pytorch.org/whl/cu102/torch-1.5.1-cp36-cp36m-win_amd64.whl
pip install https://download.pytorch.org/whl/cu102/torchvision-0.6.1-cp36-cp36m-win_amd64.whl