安装图神经网络利器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,解决

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐