error when installing pyg=2.0.4

[amiltonwong]

Hi, @rampasek ,

When I perform the installation on prerequisite packages, I got the following error when running conda install pyg=2.0.4 -c pyg -c conda-forge

(graphgps) root@milton-WS2:/data/code13/GraphGPS# conda install pyg=2.0.4 -c pyg -c conda-forge
Collecting package metadata (current_repodata.json): done
Solving environment: failed with current_repodata.json, will retry with next repodata source.
Initial quick solve with frozen env failed.  Unfreezing env and trying again.
Solving environment: failed with current_repodata.json, will retry with next repodata source.
Collecting package metadata (repodata.json): done
Solving environment: failed
Initial quick solve with frozen env failed.  Unfreezing env and trying again.
Solving environment: failed

UnsatisfiableError: The following specifications were found to be incompatible with each other:

Package cudatoolkit conflicts for:
pyg=2.0.4 -> pytorch-sparse -> pytorch-scatter -> cudatoolkit[version='10.2.*,11.3.*']
torchaudio -> cudatoolkit[version='>=11.1,<11.2']
torchvision -> pytorch[version='*,1.1.*,1.2.0.*,1.3.1.*,>=0.4',build=cpu*] -> magma[version='>=2.5.2,<2.5.3.0a0,>=2.5.4,<2.5.5.0a0'] -> cudatoolkit[version='10.0|10.0.*,10.1|10.1.*,10.2|10.2.*,9.2|9.2.*']
cudatoolkit
Package pytorch conflicts for:
pytorch=1.10

Any hints to fix this issue?

Thanks

[rampasek]

Hi @amiltonwong,

Seems it is a compatibility issue between different PyG, PyTorch and Cuda versions. For the current GraphGPS you need PyG=2.0.4 and PyTorch=1.10 (from my experience performer-pytorch package has issues with PyTorch v1.11).

Try to find a way how to install those two on your system. On my system I have PyTorch 1.10 with cudatoolkit 11.3: conda install pytorch=1.10 torchvision torchaudio cudatoolkit=11.3 -c pytorch -c nvidia

I hope that helps!