Install pytorch geometric on colab, and libcudart.so.10.2 appears when importing the package

Background problem description ：

stay google Colab Use graph neural network on the package pyg, The execution process is as follows ：

Step 1： stay Notebook Setting Set the environment as GPU;

Step 2： see Colab in torch Version information for ：

import torch

torch.__version__

notes ： When I use it, the output information is ：1.9.0+cu111

Step 3： Copy pyg The installation command in the document goes to notebook Of cell in ：

!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

Installation process smooth .

Step 4： Test and import the corresponding package ：

from torch_geometric.data import Data

here , Report errors ：libcudart.so.10.2: cannot open share object file

resolvent ：

Execute the process above Step 3 Before that, in a cell Run the following command in ：

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

And then install it pyg, Problem solved .

thank ：

Thank you for the article of manong homeland , stay Google Colab Installation on PyTorch Geometric | Code farm home

  Refer to the article address ：https://www.codenong.com/16f1ba6b9c2643d3dd12/