如果您在配置最新版PyG时遇到诸如cuda版本与pytorch版本及pyg版本不匹配的问题,在确保各个部分版本已经匹配,但仍有问题时,请用以下配置方案,(该方法为与PyG原作者讨论后得出的)。
cuda版本10.0
cudnn版本7.6.1
以下首先从源码安装了 torch-scatter、 torch-sparse、 torch-cluster、torch-spline-conv 之后安装torch-geometric 、torch 、torchvision,注意一定要从源码安装,不然就会出问题。
pip3 install torch-scatter
pip3 install torch-sparse
pip3 install torch-cluster
pip3 install torch-spline-conv
pip3 install torch-geometric torch==1.4.0+cu100 torchvision==0.5.0+cu100 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple