PyTorch 是一个流行的开源机器学习库,适用于深度学习。以下是安装 PyTorch 的步骤。
系统要求
- 操作系统:Linux、macOS 或 Windows
- Python 版本:3.6、3.7、3.8 或 3.9
安装步骤
- 打开终端或命令提示符。
- 输入以下命令:
pip install torch torchvision torchaudio
- 等待安装完成。
示例代码
安装完成后,您可以尝试以下示例代码:
import torch
print(torch.__version__)
如果您想了解更详细的安装步骤和示例代码,请访问PyTorch 官方文档。
