换源安装pytorch-gpu

PyTorch 是一个非常流行的深度学习框架,但是,官方的 PyTorch 下载地址在国外,访问速度很慢。
由于众所周知的原因,下载速度和链接很不稳定。网上给的答案都是去使用阿里云镜像,然而阿里云的镜像好久没更新了,根本没法满足现在的需求。

这里介绍一个比较冷门但质量稳定的镜像源:南京大学开源镜像站。它们一直在同步官方最新的 PyTorch 镜像源,并且速度非常快。

使用方法就是将安装命令中的 https://download.pytorch.org/whl/ 替换为 https://mirrors.nju.edu.cn/pytorch/whl/。

比如,你要安装 CUDA 12.6 的 PyTorch,官方的安装命令是:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

你只需要改成:

pip3 install torch torchvision torchaudio --index-url https://mirrors.nju.edu.cn/pytorch/whl/cu126

然后就可以了。

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐