- 安装gpu版本的pytorch,如果换源使用国内源,很可能安装完成后发现是cpu版本的,可能是因为国内源没有对应的版本,想再改成gpu版本的很麻烦,
- 使用默认源下载速度又太慢,
- 最快速的方式是确定好自己的cuda版本后,在官网查到自己要下载的torch和torchvision的版本,再去下面的链接找到对应的安装文件下载,然后在本地安装
确定对应版本
https://pytorch.org/get-started/previous-versions/
https://pytorch.org/get-started/previous-versions/