安装 torch1.7#
1. 安装好 cuda 和 cudnn#
具体可看我之前发布的文章
2. 在官网找到适合自己的版本#
Previous PyTorch Versions | PyTorch
我安装的是 cuda11 windows 系统 使用这个命令 不同的系统在这里的安装命令都不太一样,具体看官方的代码
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
没有报错就是安装完毕了,如果安装很慢请换 pip 的源