CUDA 12.1 compatibility with RTX 2000 Ada Generation, 8 GB

Dear Nvidia Community,

I am currently looking to buy a new laptop with GPU. The laptop that I am currently evaluating offers Nvidia RTX 2000 Ada Generation, 8 GB, GDDR6. I am planing to use with the OS Ubuntu 22.04.

I would like to ask if Nvidia RTX 2000 Ada Generation is compatible with CUDA 12.1? Specifically, I would like to use the GPU for Pytorch (https://pytorch.org/). Can you advise me on this GPU hardware compatibility with CUDA 12.1? Thank you.

The RTX 2000 Ada has cc 8.9 so anything cuda 11.8 and up should work.