PyTorch 是一个流行的开源机器学习库,用于应用深度学习。以下是如何在您的系统上安装 PyTorch 的步骤。

安装步骤

  1. 确定 PyTorch 版本
    首先,您需要确定要安装的 PyTorch 版本。您可以根据您的操作系统和 Python 版本来选择合适的版本。

  2. 安装 Python
    PyTorch 需要 Python 3.6 或更高版本。如果您还没有安装 Python,请前往 Python 官网 下载并安装。

  3. 安装 PyTorch
    使用以下命令安装 PyTorch:

    pip install torch torchvision torchaudio
    

    如果您需要 GPU 支持的版本,请使用以下命令:

    pip install torch torchvision torchaudio torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
    
  4. 验证安装
    安装完成后,可以通过运行以下代码来验证 PyTorch 是否已正确安装:

    import torch
    print(torch.__version__)
    

其他资源

如果您需要更多关于 PyTorch 的信息,可以访问以下链接:

PyTorch Logo