以下是在 Windows 系统中安装 PyTorch 的详细步骤：

1. 安装 Anaconda（可选但推荐）

• Anaconda 是一个用于科学计算的 Python 发行版，它可以帮助你更方便地管理 Python 环境和安装各种库。
• 你可以从 Anaconda 官方网站（https://www.anaconda.com/products/individual）下载适合你 Windows 系统版本（32 位或 64 位）的 Anaconda 安装包。
• 下载完成后，双击安装包，按照安装向导的提示进行安装。在安装过程中，你可以选择安装路径等选项。

2. 创建并激活虚拟环境（可选但推荐）

• 打开 Anaconda Prompt（如果没有安装 Anaconda，可以使用 Windows 的命令提示符）。
• 创建一个新的虚拟环境，例如，命名为 “pytorch_env”，在 Anaconda Prompt 中输入以下命令：
  • conda create -n pytorch_env python=3.8（这里以 Python 3.8 为例，你可以根据需要选择其他版本）
  
  
• 激活虚拟环境，输入：
  • 在 Anaconda Prompt 中：conda activate pytorch_env
  • 如果使用系统自带的命令提示符，需要先找到虚拟环境所在的路径，然后使用该路径下的activate.bat文件来激活，例如：C:\Users\YourUsername\Anaconda3\envs\pytorch_env\Scripts\activate.bat
  
  

3. 确定你的系统和计算平台

• PyTorch 的安装需要根据你的系统（Windows）和计算平台（CPU 或 GPU）来选择合适的安装命令。
• 如果你的电脑没有 NVIDIA GPU，或者你只想使用 CPU 来运行 PyTorch，那么安装会相对简单。
• 如果有 NVIDIA GPU 并且想利用 GPU 加速计算，你需要先安装合适的 NVIDIA 驱动程序和 CUDA 工具包。

4. 安装 PyTorch（CPU 版本）

• 如果你只想使用 CPU，在激活的虚拟环境中，使用以下命令安装 PyTorch：
  • pip install torch torchvision torchaudio
  • 这个命令会从 PyTorch 官方的 Python 包索引中下载并安装最新的 CPU 版本的 PyTorch 及其相关库（torchvision 用于计算机视觉任务，torchaudio 用于音频处理）。
  
  

5. 安装 PyTorch（GPU 版本）

• 首先，确保你的 NVIDIA GPU 支持 CUDA，并且已经安装了合适的 CUDA 工具包。一般来说，你需要安装与 PyTorch 兼容的 CUDA 版本。
• 然后，根据 CUDA 版本选择合适的安装命令。例如，对于 CUDA 11.8，在激活的虚拟环境中，使用以下命令：
  • pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
  
  
• 安装完成后，你可以通过以下简单的代码来验证 PyTorch 是否安装成功并且可以使用 GPU（如果是安装的 GPU 版本）：
  • 打开 Python 交互式环境（在激活虚拟环境后，输入python），然后输入以下代码：
  
  

收起

python

复制

torch torch__version__ torchcudais_available device torchdevice device
这段代码会打印出 PyTorch 的版本号，并且如果 GPU 可用，会显示相应的信息。