PyTorch和torch_stratter在Google Colab上使用不同的CUDA版本编译，尽管试图指定相同的版

Question

我正在谷歌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_stratter是用不同的CUDA版本编译的。PyTorch的CUDA版本为10.2，torch_stratter的CUDA为11.1。请重新安装与PyTorch安装相匹配的torch_stratter。

为什么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

看起来谷歌Colab中最新的火炬散射版本是2.0.9，比2.0.8更新。因此，当您运行命令时，它不会认为已经安装了最新版本。