Windows环境下PyTorch安装失败求助：pip无法找到torch包

直接执行pip install torch失败是因为PyTorch未在PyPI托管通用包，必须根据硬件类型（CPU/CUDA版本）使用官方指定的安装命令，具体操作如下：

1. 选择对应版本的安装命令 #

• CPU版本：无需独立显卡，直接运行：
  pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
• CUDA版本：先在命令行执行nvidia-smi，查看输出右上角的CUDA Version，再选择对应命令：
  • CUDA 12.1：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
  • CUDA 11.8：pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
    其他CUDA版本可替换命令中的cuXXX部分。

2. 验证安装成功 #

打开Python终端，执行以下代码确认：

import torch
# 检查CUDA是否可用（CPU版本输出False）
print(torch.cuda.is_available())
# 查看PyTorch版本
print(torch.__version__)

内容的提问来源于stack exchange，提问作者DevProd_