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

安装步骤

  1. 选择合适的环境:根据您的操作系统和 Python 版本,选择合适的 PyTorch 版本。
  2. 使用 pip 安装:打开命令行工具,输入以下命令进行安装:
pip install torch torchvision torchaudio
  1. 验证安装:安装完成后,您可以通过以下命令来验证 PyTorch 是否安装成功:
python -c "import torch; print(torch.__version__)"

示例代码

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

import torch

x = torch.ones(5)
y = torch.zeros(3)
print(x + y)

您可以访问我们的 PyTorch 示例代码 页面,了解更多示例。

相关链接

返回首页