对于pip的源更换,我们通常有两种策略:临时切换和永久设置。
1. 临时切换源
在执行pip install命令时,只需添加-i参数并指定国内镜像地址,如下所示:
这种方法仅在当前终端会话有效,关闭窗口后将恢复到默认源。
2. 永久更换源
为了更便捷地操作,推荐采用永久设置的方式。通过pip config命令,我们可以更改全局源地址:
要恢复默认源,只需运行:pip config unset global.index-url
针对PyTorch的CUDA版本安装,官网的pip命令可能会导致下载速度缓慢。为解决这个问题,可以尝试使用SJTU镜像源,命令如下:
这样可以确保下载CUDA版本的PyTorch,而且速度提升明显。
当然,你也可以选择其他镜像源,如阿里云镜像站。只需访问相应的CUDA版本链接,然后用pip install命令安装即可。例如,阿里云的PyTorch安装包链接可以在开源镜像站查找并使用。