Open OrangeCat7777777 opened 1 year ago
Hi! This looks like an issue caused by torch-cluster, not torchdrug. For pytorch 1.8.1, torch-cluster 1.6.0 works on my side. Maybe you can try this version.
My environment:
cudatoolkit 11.1.1
pytorch 1.8.0
torch-cluster 1.6.0
torch-scatter 2.1.1
torchdrug 0.2.1
pip install torch-cluster==1.6.0
really works for me.
But solving this problem, I have encountered this error.
AttributeError: module 'torchdrug.layers' has no attribute 'MLP'
The code is from the tutorials, the part of "Define a Graph Neural Network for Classification"
It really confuses me that when I import torchdrug.layers
and get the attribute MLP, no problems occured.
2023.7.31 reply: When I restart the jupyter kernel, the problem just goes away.
great! it is work for me!
I'm trying to run the "Property Prediction" tutorial (https://torchdrug.ai/docs/tutorials/property_prediction.html), but I encounter the following error:
My environment: