I’ve also tried pytorch 1.9.0 and CUDA 11.2

I guess you’ve build PyTorch from source using this CUDA version or did you use a pip wheel from another source (as the official binaries would either ship with CUDA10.2 (won’t work for your 3080) or CUDA11.1)?
In any case, could you post a minimal, executable code snippet which we could use to reproduce this issue, please?