search

snap-stanford / pretrain-gnns

Strategies for Pre-training Graph Neural Networks
MIT License
956 stars 161 forks source link

Docker for running code? #23

Closed cyrusmaher closed 3 years ago

cyrusmaher commented 4 years ago

Hi Weihua, I love your pre-training paper, and I'm looking forward to trying out the code. I tried installing the particular library versions that you list, but I can't get torch-scatter==1.1.2 to install. Could you share a docker file or image to run the code? I anticipate this would help others as well. Thanks for your help!

holoword commented 3 years ago

Hi Weihua, I love your pre-training paper, and I'm looking forward to trying out the code. I tried installing the particular library versions that you list, but I can't get torch-scatter==1.1.2 to install. Could you share a docker file or image to run the code? I anticipate this would help others as well. Thanks for your help!

Maybe you can download from torch-scatter GitHub release page, and use pip install the package! Meanwhile, I agree with the docker image which is a good choice to reproduce the results!

cyrusmaher commented 3 years ago

Yes, I did try pip install, but there were incompatible dependencies. Seems solvable with a separate conda environment if the docker image is not convenient.

mdanb commented 3 years ago

@cyrusmaher did you create the image by any chance?