开始

Pytorch依赖于CUDA和CuDNN，如果没有安装请参阅：win10安装CUDA和cuDNN

傻瓜式安装

pip install torch==1.7.0+cu110 -f https://download.pytorch.org/whl/torch_stable.html
pip install torchvision==0.8.1+cu110 --no-deps -f https://download.pytorch.org/whl/torch_stable.html

分别执行以上两条命令来安装torch和torchvision

如果觉得下载慢，可以尝试下面的手动下载

手动下载和安装

下载

在pytorch官方下载地址中选择合适的版本下载，根据自己的cuda版本以及python版本来确定

torch/torchvision 都需要下载

嫌麻烦的话可以查看傻瓜式安装步骤

比如你的python版本为3.7，cuda版本为11.0，就可以下这两个，当然下高版本的torch也可以

安装

命令行运行pip install xxx.whl就行了

一些问题

报错：Torch not compiled with CUDA enabled

如果前面的cuda都安装正常，大概率是因为Pytorch版本不匹配造成的，有可能为python版本不匹配，也可能是cuda版本不匹配，如果自动安装失败，请尝试手动下载