PyTorch 是一个流行的开源机器学习库,用于应用深度学习。以下是如何在您的系统上安装 PyTorch 的步骤。
安装步骤
确定 PyTorch 版本
首先,您需要确定要安装的 PyTorch 版本。您可以根据您的操作系统和 Python 版本来选择合适的版本。安装 Python
PyTorch 需要 Python 3.6 或更高版本。如果您还没有安装 Python,请前往 Python 官网 下载并安装。安装 PyTorch
使用以下命令安装 PyTorch:pip install torch torchvision torchaudio
如果您需要 GPU 支持的版本,请使用以下命令:
pip install torch torchvision torchaudio torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
验证安装
安装完成后,可以通过运行以下代码来验证 PyTorch 是否已正确安装:import torch print(torch.__version__)
其他资源
如果您需要更多关于 PyTorch 的信息,可以访问以下链接:
PyTorch Logo