问题背景描述:
在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 | 码农家园