pip - 尽管尝试指定相同的版本，但 PyTorch 和 torch_scatter 在 Google Colab 上使用不同的 CUDA 版本编译

Question

我正在 Google colab 上安装 pytorch 几何。我以前做过很多次，没有任何问题，但它突然停止工作。自从它工作以来，我没有更改我的代码。这是我的安装方式：

!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
!pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
!pip install torch-geometric

pytorch 版本应该是 1.8.1+cu102，上面的打印语句确认。我在安装时指定版本!pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html。但是，当我import torch_geometric收到错误消息时：

检测到 PyTorch 和 torch_scatter 是用不同的 CUDA 版本编译的。PyTorch 的 CUDA 版本为 10.2，torch_scatter 的 CUDA 版本为 11.1。请重新安装与您的 PyTorch 安装相匹配的 torch_scatter。

为什么torch_scatter不使用 CUDA 10.2 版进行编译？有没有办法强制它与这个版本一起编译？

Answer 1

您可以尝试指定您使用的链接提供的最新车轮版本：https ://pytorch-geometric.com/whl/torch-1.8.1+cu102.html （2021 年 11 月 22 日为2.0.8）：

pip install torch-scatter==2.0.8 -f https://data.pyg.org/whl/torch-1.8.1+cu102.html

看起来 Google Colab 中最新的 torch-scatter 版本是 2.0.9，比 2.0.8 更新。因此，当您运行命令时，它不会认为已经安装了最新版本。