In the setup of Beluga, specify the torch-geometric version to be able to install it from pypi, instead of the default CC wheelhouse:
pip install torch-geometric==1.5.0
Otherwise, by default it will install from the Compute Canada wheelhouse, which at this moment has torch-geometric 1.4.2 as the most recent version. This will downgrade all the previously installed 1.5.0 torch packages to 1.4
In the setup of Beluga, specify the torch-geometric version to be able to install it from pypi, instead of the default CC wheelhouse:
pip install torch-geometric==1.5.0
Otherwise, by default it will install from the Compute Canada wheelhouse, which at this moment has
torch-geometric 1.4.2
as the most recent version. This will downgrade all the previously installed 1.5.0 torch packages to 1.4