系统要求 📱

  • 支持平台: Windows | macOS | Linux
  • Python 版本: 3.7 - 3.10
  • CUDA 版本 (可选): 11.x 或 12.x
PyTorch_Installation

安装方法 🛠️

使用 Conda 安装 🐍

  1. 打开终端或 Anaconda Prompt
  2. 运行命令: 
    conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch
  3. 完成后验证安装: 
    import torch
print(torch.__version__)
Conda_install

使用 Pip 安装 📦

  1. 确保已安装 Python 和 pip
  2. 运行命令: 
    pip install torch torchvision torchaudio
  3. 验证安装: 
    import torch
print(torch.cuda.is_available())
Pip_install

从源码编译 🧪

  1. 安装依赖: 
    sudo apt-get install build-essential cmake libopenblas-dev
  2. 下载源码并编译: 
    git clone https://github.com/pytorch/pytorch.git
cd pytorch
python setup.py build develop
  3. 验证安装: 
    import torch
print(torch.__version__)
From_source

验证安装 ✅

  • 运行以下代码检查是否成功: 
    import torch
print(f"PyTorch 版本: {torch.__version__}")
print(f"CUDA 可用: {torch.cuda.is_available()}")
  • 若出现错误，请参考 PyTorch 官方文档 调整配置

常见问题 ❓

PyTorch_Logo