pytorch_geometric环境配置
pytorch_geometric是一个图深度学习库。除了pytorch,安装这个库还需要安装torch_cluster, torch_scatter, torch_sparse, torch_spline_conv这四个库,安装过程需要注意gcc/g++, cuda/cudnn, python, pytorch, torch_cluster/torch_scatter/torch_sparse/torch_spline_conv之间的版本对应,比较复杂繁琐,故在此记录。
环境及约束条件
- Ubuntu 20.04(无root权限,只能在用户目录下安装)
- gcc/g++ 9.3.0
- conda 4.5.4
- python 3.6.5
- GPU: Tesla V100-PCIE-32GB(驱动已经由系统管理员安装)
- 由于使用了最新的API,因此必须使用pytorch 1.10.1
各个依赖库版本
-
gcc 8.5.0, g++ 8.5.0
-
cuda 10.2, cudnn 8.3.2
-
torch 1.10.1+cu102
-
torch_cluster 1.5.9
-
torch_scatter 2.0.9
-
torch_sparse 0.6.12
-
torch_spline_conv 1.2.1
-
torch_geometric 2.0.3