Which cuda version was it tested with?

aiqm / torchani

Accurate Neural Network Potential on PyTorch

https://aiqm.github.io/torchani/

MIT License

446 stars 125 forks source link

Which cuda version was it tested with? #633

Closed UnixJunkie closed 9 months ago

UnixJunkie commented 9 months ago

And so, expected to run properly...

UnixJunkie commented 9 months ago

I suspect 11.7

isayev commented 9 months ago

Hey, @UnixJunkie, it should work with any modern CUDA. We tested everything from 9.x (a while ago) to 12. TorchANI relies on pytorch for that. However if you are interested in our custom cuaev CUDA extension you need to manually compile for that specific version you have.

UnixJunkie commented 9 months ago

Ok, thanks!