search

microsoft / Graphormer

Graphormer is a general-purpose deep learning backbone for molecular modeling.
MIT License
2.14k stars 337 forks source link

Could upgrade and support torch-geometric 2.0 ? #77

Open zsz00 opened 2 years ago

zsz00 commented 2 years ago

Could upgrade and support torch-geometric 2.0 ?

shiyu1994 commented 2 years ago

@zsz00 Thanks for using Graphormer. I've tested before with torch-geometric 2.0.3. It works with Graphormer. So I think it is okay to install torch-geometric 2.0 and use it with Graphormer.

zsz00 commented 2 years ago

yes, I tested with torch-geometric 2.0.3 , pytorch 1.10.1, cuda 11.3, nvidia 3090, ubuntu 18.04 python3.9, is ok !!!

YouJiacheng commented 2 years ago

If you do not need oc20 dataset(which use torch-geometric 1.x and not compatible with 2.x), it's okay to use 2.0.

skye95git commented 2 years ago

yes, I tested with torch-geometric 2.0.3 , pytorch 1.10.1, cuda 11.3, nvidia 3090, ubuntu 18.04 python3.9, is ok !!!

Why did you change the version? Is it because install.sh doesn't work properly?