Skip to content

Conversation

@migalkin
Copy link
Contributor

Finally managed to run TorchDrug on M2 Max, with PyTorch 1.13 should be relatively straightforward, just need to compile torch-scatter and torch-cluster from the master branch until arm64 wheels become widely available.

The CPU version works fine, the GPU version with the mps device won't work because some int64 operations used in TorchDrug are still not supported in the mps backend. Hopefully PyTorch 2.0 brings support of more tensor ops for mps

@KiddoZhu KiddoZhu merged commit e3e659d into DeepGraphLearning:master Jul 15, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants