I had the exact same issue on my new 5070ti, was able to fix it by upgrading PyTorch to CUDA 12.8. For anyone who needs it, what I ran was:

pip install -U torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu128