在 Windows 上安装 PyTorch 可以使用 pip 或 conda 管理包。
以下是使用 pip 安装 PyTorch 的步骤:
-
打开命令提示符或 PowerShell;
-
运行以下命令安装 PyTorch:
pip install torch torchvision torchaudio
如果你需要指定安装的版本,可以在命令后面通过加上 ==version 来指定,例如:
pip install torch==1.8.1 torchvision==0.9.1 torchaudio===0.8.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html
这个命令会安装 PyTorch 1.8.1、torchvision 0.9.1 和 torchaudio 0.8.1.
-
安装完成后,可以通过创建一个 Python 脚本来测试 PyTorch 是否正确安装。以下是一个简单的 TensorFlow 示例脚本:
import torch print(torch.__version__)
运行这个脚本并检查输出是否为 PyTorch 的版本号,如果输出正确则 PyTorch 安装成功。
如果您使用的是 Anaconda,可以使用以下步骤安装 PyTorch:
- 打开 Anaconda Navigator;
- 转到“Environments”选项卡,然后点击“Create”创建一个新的虚拟环境;
- 在名称字段中输入一个名称(例如“pytorch”),然后确认创建;
- 在新建的虚拟环境中,选择“Not Installed”下拉菜单的选项卡,在搜索框中输入“pytorch”,然后选择需要安装的 PyTorch 版本。点击“Apply”开始安装。
- 安装完成后,可以通过 Jupyter Notebook 或 Python 脚本来测试 PyTorch 是否正确安装。
希望这些步骤能够帮助您成功安装 PyTorch。