torch_geometric OS_Error

[fffahim]

How to solve this? Can anyone help?

[rusty1s]

Colab uses CUDA 11.1. As such, you need to download the wheels from https://data.pyg.org/whl/torch-1.9.0+cu111.html.

[manzar96]

I have a similar problem that I can not figure out how to solve.

I am using pytorch==1.9.0+cu111
and I installed pytorch-geom according to the following command as referred at the docs:
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://data.pyg.org/whl/torch-1.9.0+cu111.html
The Cuda version of the GPU is 1.4, however, I could not find the corresponding version for pytorch-geom.
What should I do?
@rusty1s
Thank you in advance.

[rusty1s]

Your system looks for CUDA 10.2, so I expect that installing from https://data.pyg.org/whl/torch-1.9.0+cu102.html should fix this error.

[manzar96]

I am using an RTX 3090 CUDA version 11.4. How is that possible?

[rusty1s]

What does torch.version.cuda return for you?

[manzar96]

What does torch.version.cuda return for you?

11.1

[rusty1s]

I see, thanks for confirming. I think you might have duplicated versions of torch-cluster installed (one for CUDA 10.2 and one for CUDA 11.1), other-wise it should never look for CUDA 10.2 in the first-place. Can you confirm that there do not exist multiple versions of torch-cluster by repeatedly running:

pip uninstall torch-cluster

and re-install torch-cluster+cu111 once again?

[manzar96]

Yes, that was the problem. Thanks a lot!