Thanks for the update!
You are using an Ampere GPU (3060), which needs CUDA>=11, while you’ve installed the PyTorch wheel with the CUDA10.2 runtime: PyTorch version: 1.9.0+cu102.
The error message also shows this error:
CUDA error: no kernel image is available for execution on the device

Install the PyTorch pip wheel or conda binary with CUDA11.1 and it should work.