Pytorch 设置国内源安装

前言

各种工具库的安装一直是我们国内技术探索路上的拦路虎，

尤其是那些源架设在国外的库，下载安装包的时候即时kb/s的速度简直让人窒息。

 

最近安装pytorch的时候也发现这个问题，下载速度不快，

虽然pytorch的下载速度已经能达到200+kb/s，但也远比不上用国内源来的快。

搜了一下发现Pytorch也可以换源安装。 目前只有阿里云上有Pytorch的镜像源文件：

https://mirrors.aliyun.com/pytorch-wheels/

下图中的是cuda版本：

 

指定阿里云镜像源的pytorch安装命令如下：

pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu118

需要注意的是，这里pip命令指定url用到的参数是-f，而不是常用的-i或--index-url

用pip install -h看看这两个参数的区别:

可以看到-f更偏向于从指定目录中查找指定包，并进行安装。

 

The End

---

 

技术之外，欢迎关注我认知成长的公众号 ：零度的成长记录