解决方案:在对自己的网络环境有绝对的自信的情况下,在PyTorch官网的pip3安装命令后直接添加 --default-time=10000 将超时时间延长到10秒。For example,在conda环境中直接执行:
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 --default-timeout=10000
背景情况:人在国外,刚下飞机,对校园网络的稳定性相当自信。Ubuntu 20.04 在按照PyTorch官网指令安装时无论使用 conda/pip3 都会因 Connection Broken 被强制终止。
推测原因:伴随着PyTorch和CUDA的不断更新迭代,安装包体积越来越大,由于一些不可名状的网络原因造成了高延迟,因此需要手动延长系统延时中止的判断。
PS: 在网络相对不稳定的情况下,可以使用清华、阿里云等第三方channels进行下载。