← Home

pytorch 镜像

8 August, 2023

pytorch 仅有一个 wheels 下载站 https://download.pytorch.org/whl ,当然 pip 是可以识别 find-links 类型的,但是在国内的网络环境下,下载却非常慢,尝试用上海交大的镜像,能临时解决问题,不过我们要在项目中构建 Docker 镜像,还是用免费的 nexus 私服构建代理比较合适。

而 nexus 私服的 pypi 只支持代理 pypi 的上游,不支持 find-links 规范,故曲线救国,先手动把 whl 包下载下来,再用 twine 上传/发布到私服。

先到华中科大的镜像站下载需要的 whl 包 , 留意下载地址的那个参数,必传否则跳转到官方,可能机器学习大热华中科大的流量也是比较惊人吧。

  1. 在浏览器的 console 执行下面脚本,拷贝所有 linux 的发行包(按照你的需要调整) ```
copy($$('a').map(it=>it.href).filter(it=>it.indexOf('linux')>0).join('\n'))
  1. 配置 ~/.pypirc
[distutils]
index-servers =
    nexus-torch-cu118

[nexus-torch-cu118]
repository = https://nexus-server/repository/pytorch-cu118/
username = user
password = pass
  1. 安装好 twine ,挨个包执行上传/发布
twine upload --repository nexus-torch-cu118 torch-2.0.0+cu118-cp39-cp39-linux_x86_64.whl
  1. 记得在 pypi group 里加上发布的那个 self-host 仓库