CUDA 12.1环境下无法安装Torch系列组件，求解决方案

1. 检查Python版本兼容性 #

PyTorch对Python版本有严格要求，CUDA 12.1对应的PyTorch（含nightly版本）仅支持Python 3.8~3.11版本，先确认你的Python版本：

python --version

若版本不在范围内，建议降级或升级到兼容版本。

2. 确认系统架构与CUDA环境有效性 #

• 确保系统为x86_64架构（ARM架构如M1/M2暂不支持CUDA版PyTorch），执行命令检查：

uname -m

• 验证CUDA 12.1是否正确安装并配置环境变量：

nvcc --version

若输出版本不是12.1，需先修复CUDA的安装或环境变量配置。

3. 更换安装方式规避网络或索引问题 #

方法一：添加国内镜像源加速安装 #

在原命令后追加国内PyPI镜像源，解决网络导致的包拉取失败：

pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121 -i https://mirrors.aliyun.com/pypi/simple/

方法二：手动下载whl文件本地安装 #

1. 前往PyTorch官方whl仓库（cu121 nightly分区），筛选与你的Python版本、系统匹配的torch、torchvision、torchaudio文件（文件名含cp38/cp39等对应Python版本，linux_x86_64或win_amd64对应系统）。
2. 下载完成后执行本地安装：

pip install ./torch-xxx-cu121.whl ./torchvision-xxx-cu121.whl ./torchaudio-xxx-cu121.whl

4. 更新pip版本 #

旧版pip可能不支持新的包索引格式，先升级pip：

pip install --upgrade pip

之后重新执行PyTorch安装命令。

5. 验证虚拟环境配置（若使用） #

如果在虚拟环境中操作，确保虚拟环境已正确加载Python版本和CUDA环境变量，退出并重新激活虚拟环境后再尝试安装。

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