PyTorch 是一个流行的开源机器学习库,适用于 Python 程序。以下是 PyTorch 在 Python 环境下的安装步骤。

安装步骤

  1. 安装 Python:首先确保你的系统中已经安装了 Python。推荐使用 Python 3.6 或更高版本。

  2. 创建虚拟环境:为了更好地管理项目依赖,建议创建一个虚拟环境。

    python -m venv myenv
    source myenv/bin/activate  # 在 Windows 上使用 myenv\Scripts\activate
    
  3. 安装 PyTorch:使用以下命令安装 PyTorch。

    pip install torch torchvision torchaudio
    

    你也可以使用以下命令安装 CPU 版本的 PyTorch。

    pip install torch torchvision torchaudio cpuonly
    
  4. 验证安装:安装完成后,你可以通过以下命令验证 PyTorch 是否已正确安装。

    import torch
    print(torch.__version__)
    

其他资源

如果你需要更多关于 PyTorch 的信息,可以访问我们的PyTorch 教程页面

示例图片

以下是一些 PyTorch 相关的图片示例:

PyTorch Logo
PyTorch Network