在这里插入图片描述
评论区看到的答案,非常好用!!!哈哈哈哈哈哈

前言

pytorch的cpu的包可以在国内镜像上下载,但是gpu版的包只能通过国外镜像下载,网上查了很多教程,基本都是手动从先将gpu版whl包下载下来,然后再手动安装,如何最快的通过pip的命令安装呢?下面我细细讲下。

解决办法

目前国内有pytorch的gpu版的whl包只有阿里云上的:https://mirrors.aliyun.com/pytorch-wheels/
在这里插入图片描述

这里以11.3为例介绍如何安装

CUDA 11.3

pip install torch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 -f 
https://mirrors.aliyun.com/pytorch-wheels/cu113

在这里插入图片描述
可以看到下载还是很快的,官网只有200k。

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐