换源安装 torch-gpu
·
换源安装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
然后就可以了。
更多推荐




所有评论(0)