PyTorch with CUDA 11 compatibility

In case you are seeing “random” CUDA errors, it could point to a setup issue, but usually the error is raised from the code itself. Could you post an executable code snippet, which would reproduce the issue, please?