在 windows 下如何安装 pytorch？_windows装pytorch

在 Windows 上安装 PyTorch 可以使用 pip 或 conda 管理包。

以下是使用 pip 安装 PyTorch 的步骤：

1. 打开命令提示符或 PowerShell；

2. 运行以下命令安装 PyTorch：

   pip install torch torchvision torchaudio
   1

   如果你需要指定安装的版本，可以在命令后面通过加上 ==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
   1

   这个命令会安装 PyTorch 1.8.1、torchvision 0.9.1 和 torchaudio 0.8.1.

3. 安装完成后，可以通过创建一个 Python 脚本来测试 PyTorch 是否正确安装。以下是一个简单的 TensorFlow 示例脚本：

   import torch
   print(torch.__version__)
   1
   2

   运行这个脚本并检查输出是否为 PyTorch 的版本号，如果输出正确则 PyTorch 安装成功。

如果您使用的是 Anaconda，可以使用以下步骤安装 PyTorch：

1. 打开 Anaconda Navigator；
2. 转到“Environments”选项卡，然后点击“Create”创建一个新的虚拟环境；
3. 在名称字段中输入一个名称（例如“pytorch”），然后确认创建；
4. 在新建的虚拟环境中，选择“Not Installed”下拉菜单的选项卡，在搜索框中输入“pytorch”，然后选择需要安装的 PyTorch 版本。点击“Apply”开始安装。
5. 安装完成后，可以通过 Jupyter Notebook 或 Python 脚本来测试 PyTorch 是否正确安装。

希望这些步骤能够帮助您成功安装 PyTorch。