PyTorch y torch_scatter fueron compilados con diferentes versiones de CUDA en Google Colab, a pesar de intentar especificar la misma versión.

Estoy instalando pytorch geometric en Google Colab. Lo he hecho muchas veces antes y no he tenido problemas, pero de repente ha dejado de funcionar. No he cambiado mi código desde que funcionaba. Así es como lo instalo:

!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

La versión de pytorch debe ser 1.8.1+cu102, confirmado por la declaración de impresión anterior. Especifico la versión al instalar con !pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html. Sin embargo, cuando importo torch_geometric obtengo el error:

Se ha detectado que PyTorch y torchscatter se compilaron con diferentes versiones de CUDA. PyTorch tiene la versión CUDA 10.2 y torchscatter tiene la versión CUDA 11.1. Por favor, reinstale torch_scatter que coincida con su instalación de PyTorch.

¿Por qué torch_scatter no se está compilando con la versión CUDA 10.2? ¿Existe alguna manera de forzarlo a compilar con esta versión?