温馨提示×

CentOS上PyTorch的安装包在哪下载

小樊
58
2025-08-16 23:57:36
栏目: 智能运维

在CentOS上安装PyTorch,可通过pip或conda方式,安装包下载地址及命令如下:

  • pip方式
    • CPU版本:无需指定额外索引,直接使用命令pip3 install torch torchvision torchaudio,安装包会从PyTorch官方默认源下载。
    • GPU版本:需指定CUDA版本对应的索引URL,如CUDA 11.3版本,命令为pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113,可从PyTorch官网获取最新安装命令及对应版本下载链接。
  • conda方式
    • CPU版本:命令为conda install pytorch torchvision torchaudio cpuonly -c pytorch,安装包从PyTorch的conda官方频道下载。
    • GPU版本:例如CUDA 11.3版本,命令是conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch,同样从PyTorch的conda官方频道获取包。

0