Doesn't recognize my RTX 6000 mobile

[orionflame]

Hi,

I am using RTX 6000 mobile (24GB) on my laptop but it doesn't recognize it. Says "Unfortunately, there is no compatible GPU available to support your training."

Why is that? How can I fix this?

My GPU works in stable diffusion for example and quite powerful to handle it, that's why I am wondering how to fix it.

I saw a post about uninstalling and installing pytorch etc with different config but it was recommended for 3000 series. I couldn't find anything for RTX 6000, not RTX A6000.