praggya:

RuntimeError: CUDA error: no kernel image is available for execution on the device


This would indicate that the currently used PyTorch build doesn’t contain kernels for the right compute capability (8.6 in your case), which would happen if you’ve installed e.g. the CUDA10.2 binaries. Make sure to select the CUDA11.1 (or 11.3) runtime during the installation.