在Colab上安装pytorch-geometric，导入包时出现libcudart.so.10.2

问题背景描述：

在google Colab上使用图神经网络的包pyg，执行过程如下：

Step 1： 在Notebook Setting里设定环境为GPU；

Step 2： 查看Colab里torch的版本信息：

import torch

torch.__version__

注：我使用时输出信息是：1.9.0+cu111

Step 3： 复制pyg文档里的安装命令到notebook的cell里：

!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

安装过程顺利。

Step 4： 测试导入相应的包：

from torch_geometric.data import Data

此时，报错：libcudart.so.10.2: cannot open share object file

解决方法：

在上面的执行过程的Step 3之前先在一个cell里运行如下命令：

!export CPATH=/usr/local/cuda/include:$CPATH

然后再安装pyg，问题得到解决。

致谢：

感谢码农家园的文章，在Google Colab上安装PyTorch Geometric | 码农家园

 参考文章地址：https://www.codenong.com/16f1ba6b9c2643d3dd12/