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

【问题标题】：PyTorch and torch_scatter were compiled with different CUDA versions on Google Colab despite attempting to specify same version尽管尝试指定相同的版本，但 PyTorch 和 torch_scatter 在 Google Colab 上使用不同的 CUDA 版本编译
【发布时间】：2022-01-03 20:39:10
【问题描述】：

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

!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 版一起编译？有没有办法强制它用这个版本编译？

【问题讨论】：

标签： pip pytorch pytorch-geometric

【解决方案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 更新。因此，当您运行您的命令时，它不会认为已经安装了最新版本。

【讨论】：