Google Colab运行PyTorch Geometric代码报libcudart.so.10.2不存在如何解决

• 2021年10月Google Colab官方完成运行环境迭代，预装的CUDA运行时版本从10.2升级为更高版本，系统默认不再保留CUDA 10.2的动态链接库文件libcudart.so.10.2
• 你此前安装的PyTorch Geometric（PyG）及相关依赖包均为适配CUDA 10.2编译的版本，执行from torch_geometric.datasets import TUDataset导入操作时，会触发PyG底层依赖的加载逻辑，主动调用CUDA 10.2的核心动态库，系统找不到对应文件就会触发该OSError

按照以下步骤操作即可修复该问题：

• 第一步：卸载所有旧版本的PyG相关组件，避免新旧版本冲突

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

• 第二步：查询当前Colab环境的PyTorch版本和CUDA版本，执行以下代码即可查看：

# 查看PyTorch版本
import torch
print(torch.__version__)

# 查看CUDA版本
!nvcc --version

• 第三步：安装适配当前环境的PyG组件，以2021年10月Colab默认的PyTorch 1.9.0 + CUDA 11.1环境为例，执行以下命令即可：

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

如果你查询到的PyTorch和CUDA版本和上述示例不同，只需要将命令末尾的版本标识替换为你查询到的对应版本即可，比如PyTorch 1.10.0 + CUDA 11.3环境，就将torch-1.9.0+cu111替换为torch-1.10.0+cu113

内容的提问来源于stack exchange，提问作者Cy-Phy