网络环境极差时，如何通过pip3带CUDA安装PyTorch？

关于pip断点续传的说明 #

pip从20.0版本开始原生支持断点续传功能，但需要满足两个前提：

• pip版本≥20.0
• 下载源服务器支持HTTP Range请求（PyTorch官方下载源https://download.pytorch.org/whl/cu132是支持的）

之前下载中断后需从头开始，大概率是pip版本过低导致续传功能未生效。

具体解决办法 #

• 升级pip到最新版本
  先执行升级命令确保续传功能正常：

  pip3 install --upgrade pip
  

  升级后重新执行原安装命令，pip会自动接续上次未完成的下载进度。

• 手动下载whl包后本地安装
  用支持断点续传的工具（如wget或curl）单独下载所需的whl文件：

  • wget命令（替换版本号为你需要的对应版本）：

    wget -c https://download.pytorch.org/whl/cu132/torch-2.x.x+cu132-cp3x-cp3x-linux_x86_64.whl
    wget -c https://download.pytorch.org/whl/cu132/torchvision-0.x.x+cu132-cp3x-cp3x-linux_x86_64.whl
    

  • 下载完成后执行本地安装：

    pip3 install torch-2.x.x+cu132-cp3x-cp3x-linux_x86_64.whl torchvision-0.x.x+cu132-cp3x-cp3x-linux_x86_64.whl
    

• 优化pip下载配置
  除增加超时和重试次数，还可通过环境变量增强稳定性：

  # 设置1小时超时时间，并开启超时断点续传
  export PIP_DOWNLOAD_TIMEOUT=3600
  export PIP_BREAKPOINT_ON_TIMEOUT=true
  # 重新执行安装命令
  pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132 --retries=100
  

• 指定pip缓存目录
  pip默认会缓存下载文件，新版pip能识别缓存中的未完成文件并续传。若需自定义缓存目录，可添加--cache-dir参数：

  pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu132 --timeout=3600 --retries=100 --cache-dir ~/custom_pip_cache
  

内容的提问来源于stack exchange，提问作者HMPtwo