在这个教程中,我们将向您展示如何在 Python 中安装 PyTorch。PyTorch 是一个流行的开源机器学习库,广泛用于深度学习。

安装步骤

  1. 检查 Python 版本 确保您的 Python 版本至少为 3.6。您可以使用以下命令检查 Python 版本:

    python --version
    

    检查 Python 版本

  2. 创建虚拟环境 创建一个虚拟环境可以避免与系统中的其他项目发生冲突。您可以使用以下命令创建虚拟环境:

    python -m venv myenv
    

    创建虚拟环境

  3. 激活虚拟环境 在 Windows 上,您可以使用以下命令激活虚拟环境:

    myenv\Scripts\activate
    

    在 macOS/Linux 上,您可以使用以下命令激活虚拟环境:

    source myenv/bin/activate
    

    激活虚拟环境

  4. 安装 PyTorch 您可以使用以下命令安装 PyTorch:

    pip install torch torchvision torchaudio
    

    安装 PyTorch

  5. 验证安装 安装完成后,您可以使用以下命令验证 PyTorch 是否已正确安装:

    import torch
    print(torch.__version__)
    

    验证 PyTorch 安装

下一步

现在您已经成功安装了 PyTorch,您可以开始构建自己的深度学习模型了。如果您需要更多帮助,请访问我们的 深度学习教程

深度学习教程