安装图神经网络利器Torch_geometric

最新推荐文章于 2024-08-05 14:35:27 发布

东方小烈

最新推荐文章于 2024-08-05 14:35:27 发布

阅读量409

点赞数

分类专栏：图神经网络文章标签：神经网络图神经网络

本文链接：https://blog.csdn.net/dongfangxiaozi_/article/details/108309147

版权

图神经网络专栏收录该内容

1 篇文章 0 订阅

订阅专栏

安装图神经网络利器Torch_geometric遇到的问题：

本来用的cuda是10.1

nvcc --version

Cuda compilation tools, release 10.1, V10.1.168

PyTorch使用的是1.6.0

按照教程应该安装$ pip install torch-scatter==latest+cu101 -f https://pytorch-geometric.com/whl/torch-1.6.0.html 
可以却遇到了：Torch使用的是10.2，torch-scatter 是10.1的错误提示，
可能是这个安装引导程序有些问题，所以，安装torch-scatter 10.2即可，如下：

$ pip install torch-scatter==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-sparse==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-cluster==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-spline-conv==latest+cu102 -f https://pytorch-geometric.com/whl/torch-1.6.0.html
$ pip install torch-geometric

又现实torchvision版本的问题，重新安装torchvision，解决