如何在Windows 10上安装支持CUDA的pytorch 1.2.0？

Question

我尝试在 Windows 10 上使用 Python 3.9 安装支持 CUDA 的 pytorch 1.2.0，但一直出错。

例如，我做

pip install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

并得到：

ERROR: Could not open requirements file: [Errno 2] No such file or directory: 'pytorch'

有什么想法吗？谢谢。

Answer 1

我的回答：

pip install wheel

然后使用此处的版本链接： https://www.python.org/downloads/release/python-370/

例如，如果您想要 Python 3.7 Windows 的 1.1.0，请执行以下操作：

pip install https://download.pytorch.org/whl/cu100/torch-1.1.0-cp37-cp37m-win_amd64.whl

谢谢大家

Answer 2

pip install torch==2.3.0 试试这个