Google Colab GPU环境导入PyTorch Geometric提示libcudart.so.10.2缺失

当前环境CUDA版本为11.2，已安装的PyTorch Geometric相关依赖（torch_sparse、torch_scatter等）是基于CUDA10.2版本编译的，版本不匹配导致运行时找不到对应CUDA动态库。

• 第一步：卸载所有旧版本PyG相关依赖

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

• 第二步：查询当前环境PyTorch与CUDA版本

import torch
print(torch.__version__)
print(torch.version.cuda)

• 第三步：安装和当前版本匹配的PyG依赖
  根据上一步输出的版本信息，执行对应版本的安装命令即可。例如如果输出PyTorch版本为1.12.0、CUDA版本为11.3，执行以下命令：

!pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.12.0+cu113.html

• 第四步：重新导入所需模块即可正常运行

import torch
from torch_geometric.datasets import TUDataset

Colab环境会不定期更新预装的CUDA与PyTorch版本，此前运行正常是因为当时环境版本和依赖版本匹配，版本更新后重新安装匹配版本的依赖即可解决问题。

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