安装PyTorch (Windows) - Shell/Bash

PyTorch是一个基于Python的科学计算包，主要针对两类人群：

• 第一类是对计算机视觉等领域的用户，这类用户通常已经有深度学习背景，想要在研究领域中尝试新的想法；
• 第二类人群是希望从事深度学习研究的新手，他们希望能通俗易懂地理解代码背后的机理。

在本文章中，我们将会介绍如何在Windows操作系统中使用Shell/Bash命令行工具安装PyTorch。

步骤

1. 确认系统环境变量

在开始使用PyTorch之前，需要确保您的Windows操作系统已安装最新版本的Microsoft Visual C++和CUDA驱动程序以及相关环境变量。如果您还没有安装这些必要的环境变量，请先从以下链接中下载和安装：

• Microsoft Visual C++
• CUDA驱动程序

2. 打开Shell/Bash命令行工具

为了开始安装PyTorch，您需要打开Shell/Bash命令行工具。在Windows系统中，您可以通过按下键盘上的Win + R组合键并输入cmd或Powershell打开命令行终端。

3. 安装PyTorch

在Shell/Bash命令行工具中输入以下命令来下载并安装最新版本的PyTorch：

pip install pytorch

在下载和安装完成后，您可以使用以下命令来验证是否已成功安装：

import torch
print(torch.__version__)

如果未出现任何错误，PyTorch的版本号将会被打印在命令行终端中。

结论

通过本文章提供的简要步骤，您可以轻松地在Windows操作系统上使用Shell/Bash命令行工具安装并验证最新版本的PyTorch。如果想要了解更多有关PyTorch的详细信息，请访问PyTorch官方网站。