文章目录

Background

使用以下命令下载 PyTorch 时,发现其中 cudatoolkit 包的下载速度极慢:

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

Solution

配置清华镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/
conda config --add https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/

并去掉 PyTorch 安装命令中,最后的 -c pytorch

conda install pytorch torchvision torchaudio cudatoolkit=10.2
Logo

欢迎来到FlagOS开发社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐