search

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
MIT License
2.61k stars 367 forks source link

Pip install downloading torch 2.0 #218

Closed liamtoran closed 1 year ago

liamtoran commented 1 year ago

Hi there,

After following the installation steps with pip install, I noticed that it installed pytorch 2.0, which wasn't compatible with the library. I received a missing cuda operator error while trying to load the chickenpoxdataset.

To fix the problem, I just did pip install torch==1.13. After that I could load the chickenpox dataset without any issues.

I suggest adding torch<2.0.0 or the required versions to setup.py to prevent similar issues for other users. I'm not familiar enough with the project to create a pull request myself, so I hope you can take care of it.

Thanks for your help!