PyTorch Geometric(PyG)环境配置问题解决

如果您在配置最新版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

原创文章 32 获赞 4 访问量 7638

猜你喜欢

转载自blog.csdn.net/yrwang_xd/article/details/105936721
今日推荐