NVIDIA GeForce RTX 3090 with CUDA capability sm_86 is not compatible with the current PyTorch installation

Yes, you are right. Did mistake. I’ve fixed it. It works now. Thanks for replying.