pyg_lib module installation not available

[JasonMts]

Canary models sage, gcn and gat attempt to install the pyg_lib module during their installation process but their requirements files do not include the link for the custom wheel which is needed as noted in the pyg-lib module’s README, or hard code to torch-2.1.0+cpu, which makes them fail for other versions.

This causes the following error:

$ python benchmark/install.py sage --canary -v
checking packages numpy, torch, torchvision, torchaudio are installed, generating constaints...OK
running setup for /home/user/benchmark/torchbenchmark/canary_models/sage...decompressing input tarball: /home/user/benchmark/torchbenchmark/data/Reddit_minimal.tar.gz...OK
Looking in indexes: ...api/pypi/pypi-virtual/simple, https://pypi.python.org/simple/ 
ERROR: Could not find a version that satisfies the requirement pyg_lib (from versions: none)
ERROR: No matching distribution found for pyg_lib