离线安装torch

离线安装torch 一般的:安装torch只做计算机视觉任务,则只需要Pytorch、Torchvision,如果需要处理声音,则还需要Torchaudio

离线安装torch

一般的:安装torch只做计算机视觉任务,则只需要Pytorch、Torchvision,如果需要处理声音,则还需要Torchaudio

  • 注意!!!CUDA驱动和CUDAToolkit所对应的CUDA版本都是可以向下兼容的,所以一般要选取最新的驱动和工具包安装

  • PS:一般情况下,Torch版本、CUDA版本、Python版本都是严密对应的。但是一般最新的Torch版本所对应的CUDA版本都是要低一点的。这个不用在意。需要注意的是,Python版本一定要匹配!

Torch、Torchvision、TorchAudio历史版本:
https://download.pytorch.org/whl/torch_stable.html

找到你对应的torch版本,cuda版本,python版本,操作系统版本

下载离线包,然后执行

pip install 离线包地址.whl

即可离线安装Torch、Torchvision,Torchaudio

例如:

cd C:\Users\Arix\Downloadspip install torch-1.13.0+cu117-cp37-cp37m-win_amd64.whlpip install torchvision-0.14.0+cu117-cp37-cp37m-win_amd64.whlpip install torchaudio-0.13.0+cu117-cp37-cp37m-win_amd64.whl
  • 注意,要严格按照顺序安装,pip会自动安装部分依赖包(numpy之类)