feat: enable cuequivariance kernels on the base image

[avelinoapheris]

• Modify the base image to install the cuequivariance kernels. This also involve upgrading the base cuda version.
• Add checks to ensure that the cuequivariance kernels are not loaded when using a setup without GPUs.

[jnwei]

Hi @avelinoapheris thank you for the PR. Thank you for the fix regarding checking if cuEquivariance is installed.

Regarding updating the dockerfile - do you know if this new docker recipe would be compatible on GPUs that might have an older CUDA driver (<12.5) or are otherwise incompatible (e.g. NVIDIA/cuEquivariance#209)?

One option could be to support multiple docker recipes. We had a previous PR contribution for a dockerfile specifically for Blackwell GPUs