PyTorch与torch-sparse CUDA版本不匹配问题求助

核心问题：PyTorch和torch-sparse必须使用完全一致的CUDA版本编译，版本不匹配会直接触发RuntimeError。以下是针对性的解决步骤：

1. 彻底清理现有环境 #

先卸载所有相关依赖包，避免残留冲突：

pip uninstall -y torch torch-sparse torch-geometric

同时清理pip缓存，防止安装旧版本包：

rm -rf ~/.cache/pip

2. 安装版本严格匹配的依赖 #

根据你使用的PyTorch 1.7.1，选择以下方案之一：

方案A：沿用PyTorch自带的CUDA 10.2 #

先安装带CUDA 10.2的PyTorch 1.7.1：

pip install torch==1.7.1+cu102 torchvision==0.8.2+cu102 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

再安装对应CUDA 10.2的torch-sparse 0.6.8：

pip install torch-sparse==0.6.8 -f https://pytorch-geometric.com/whl/torch-1.7.1+cu102.html

方案B：适配集群的CUDA 11.x（注意PyTorch 1.7.1最高支持CUDA 11.0） #

先安装带CUDA 11.0的PyTorch 1.7.1：

pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

再安装对应CUDA 11.0的torch-sparse 0.6.8：

pip install torch-sparse==0.6.8 -f https://pytorch-geometric.com/whl/torch-1.7.1+cu110.html

方案C：使用纯CPU版本（无CUDA） #

如果不需要GPU加速，必须同时安装CPU版的PyTorch和torch-sparse：

pip install torch==1.7.1+cpu torchvision==0.8.2+cpu torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
pip install torch-sparse==0.6.8 -f https://pytorch-geometric.com/whl/torch-1.7.1+cpu.html

你之前尝试失败很可能是只装了CPU版PyTorch，但torch-sparse仍装了CUDA版本，导致版本不匹配。

3. 验证安装结果 #

运行以下代码确认版本一致：

import torch
import torch_sparse
print(f"PyTorch CUDA版本: {torch.version.cuda}")
print(f"torch-sparse CUDA版本: {torch_sparse.__cuda_version__}")

输出的两个版本必须完全相同，才算安装成功。

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