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

确定对应版本

https://pytorch.org/get-started/previous-versions/

下载文件

https://download.pytorch.org/whl/torch_stable.html

Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐