温馨提示×

Linux版PyTorch的安装包在哪里下载

小樊
61
2025-06-17 09:53:53
栏目: 智能运维

在Linux系统上安装PyTorch有多种方法,以下是一些常用的下载链接和安装步骤:

使用pip安装PyTorch

  • 基本CPU版本
    pip install torch torchvision torchaudio
    
  • CUDA 11.3版本
    pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
    

使用conda安装PyTorch

  • 适合使用Anaconda环境的用户
    conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
    

从源码编译PyTorch

  • 适合需要自定义PyTorch功能的用户
    git clone --recursive https://github.com/pytorch/pytorch
    cd pytorch
    pip install -r requirements.txt
    python setup.py install
    

请注意,具体的安装命令可能会根据你的CUDA版本和系统环境有所不同。建议访问PyTorch官方网站获取最新的安装指南。

0