最近在使用服务器配置pytorch环境的时候,遇到了极为恼火的事情,使用pytorch官方的命令来下载GPU版本的pytorch总会是不是下载到一半就会崩溃,然而pip下载并不会断点续传(什么时候能出这个功能啊喂!)。每次下到一半失败后,又要重新下载。。。。如果加上国内源,下载的又是CPU版本。本笨比碰壁多次后,也算是找到了一些经验,在这里和大家分享
一、我们经常使用以下命令下载pytorch
pip3 install torch --index-url https://download.pytorch.org/whl/cu118
二、但是有时候会出现下面这种恼火的情况(下面这张图我直接贴了别人的window下的图,但是我实际上使用linux也是报以下的错误,这里只分享linux下的解决方案)
三、难道我们只能一直默认从头装吗?不是的,熟悉linux的朋友应该知道wget命令,我们使用wget+pip本地安装的方式来解决(wget有断点续传,相对于pip来说更坚挺,不容易崩溃)
四、wget + pip本地化安装
仔细观察上述报错,在第一行是不是有(观察你的安装pytorch的输出,应该也是有的,无论安装什么包)
这里我们可以知道,pip从上面这个网址下载包,这是通过https协议下载的。我们知道wget可以通过https下载,并且有断点续传功能。那我们就想到了先用wget下载到本地,再用pip安装的想法。
wget -c https://download.pytorch.org/whl/cu113/torch-1.12.0%2Bcu113-cp310-cp310-linux_x86_64.whl
pip install torch-1.12.0+cu113-cp310-cp310-linux_x86_64.whl
其中,-c 表示wget启用断点续传,如果你提前下了部分文件,wget会接着下载而不是重新下载。
用wget下载到本地后,可以查看一下文件名,用pip安装
五、小结
以上的方法可以拓展到所有的python包,不一定针对pytorch;遇到某个包下载到一半就失败的情况,就可以尝试用wget+pip的反射光hi