本教程将指导您如何配置 PyTorch 环境，以便在您的计算机上开始使用 PyTorch 进行机器学习。

系统要求

  • 操作系统：Windows、macOS 或 Linux
  • Python：Python 3.6 或更高版本

安装 PyTorch

  1. 访问 PyTorch 官方网站。
  2. 选择合适的 PyTorch 版本和安装选项。
  3. 根据您的操作系统和 Python 版本，下载并安装 PyTorch。

验证安装

安装完成后，您可以通过以下命令验证 PyTorch 是否已成功安装：

python -c "import torch; print(torch.__version__)"

配置 CUDA（可选）

如果您想使用 CUDA 加速 PyTorch，您需要安装 CUDA 驱动程序和 cuDNN 库。

  1. 访问 NVIDIA 官方网站 下载并安装 CUDA 驱动程序。
  2. 下载并安装 cuDNN 库。
  3. 将 cuDNN 库的路径添加到您的系统环境变量中。

示例代码

以下是一个简单的 PyTorch 示例代码：

import torch

# 创建一个随机张量
x = torch.randn(5, 3)

# 打印张量
print(x)

扩展阅读

如果您想了解更多关于 PyTorch 的信息，请访问以下链接：

PyTorch Logo