🐛 [BUG] Does not work with RTX 4080 GPU about nequip HOT 6 CLOSED

Hi @vasalx, can you share which PyTorch and CUDA versions you'd need for the RTX 4080?

from nequip.

Hi @simonbatzner, thanks for getting back so fast.

As described in pytorch/pytorch#91973, stable PyTorch 1.13.x with CUDA 11.6 or 11.7 should be good. Not sure if this is relevant, but cuFFT works only with CUDA 11.8 which requires a nightly build of PyTorch.

This is not exclusive to 4080, though. All higher end 40xx NVIDIA GPUs seem to be affected, and possibly even 3090.

from nequip.

@vasalx we are affected by upstream issues that exist in 1.13 (but not in 1.11) that may affect you if you use PyTorch!=1.11 (see #311), but besides these PyTorch bugs we do support 1.13. What problem are you having, exactly?

from nequip.

Hi, @Linux-cpp-lisp! How do I make it work with the latest then? When I try to update existing conda env, I start to get CUDA errors (do not have them handy, but will try to replicate if needed). In new env with preinstalled latest PyTorch and CUDA, trying to install NequIP degrades those to the requirements of the package (PT 1.11).

from nequip.

@Linux-cpp-lisp, never mind. I see that it is possible on the develop branch. Sorry for troubling you!

from nequip.

No worries, thanks for your interest in our code @vasalx ! I'm planning to merge and release develop soon.

from nequip.