The right Pytorch for GeForce 3090 & CUDA 11.1

This might help: