国内下载torch
的速度很慢,那么我们可以先去到官方网站手动下载whl
,下载的时候可以使用多线程下载器IDM
下载,这样的大大提升下载速度,基本上就几分钟就可以下载完成,然后可以使用pip install
命令安装whl
安装包。
这里贴出下载torch
的地址。
https://download.pytorch.org/whl/torch/
这里需要注意,下载的torch
版本一定要符合你本机中的环境,比如,我这里是python
版本的是3.9
,cuda
版本是11.8
,并且是Windows
系统,那么你可以在上面的网站中搜索找到符合你本机的版本。
Q.E.D.