Google Colab安装PyTorch Geometric时torch_scatter CUDA版本不匹配

问题原因 #

• 你使用的-f参数仅为pip提供额外的包查找路径，并未禁用默认的PyPI源。pip默认会优先安装版本号更高的包，忽略CUDA版本匹配规则。近期PyPI平台的torch_scatter等依赖包更新了版本，默认编译的CUDA版本为11.1，pip优先拉取了该版本，才会出现和你本地PyTorch 1.8.1+cu102不匹配的报错。

修复方案 #

首先执行如下命令清除已安装的错误版本依赖，避免残留影响：

!pip uninstall -y torch-scatter torch-sparse torch-cluster torch-spline-conv

之后两种方案任选其一重新安装即可：

方案1：强制仅从PyG官方源拉取包 #

在每个依赖安装命令后添加--no-index参数，禁用PyPI源，只从你指定的CUDA102对应的源查找安装包：

!pip install torch==1.8.1 torchvision torchtext
import torch; print(torch.__version__); print(torch.version.cuda)

!pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html --no-index
!pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html --no-index
!pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html --no-index
!pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html --no-index
!pip install torch-geometric

方案2：手动锁定安装包版本 #

直接指定带CUDA版本后缀的包名，完全锁定安装的版本和编译环境：

!pip install torch-scatter==2.0.8+cu102 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-sparse==0.6.12+cu102 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-cluster==1.5.9+cu102 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-spline-conv==1.2.1+cu102 -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-geometric

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