after i upgrade my CUDA version from 11.8 to 12.1, my python codes stopped working and throwing error like : RuntimeError: CUDA error: device-side assert triggered
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.
/opt/conda/conda-bld/pytorch_1708025847094/work/aten/src/ATen/native/cuda/IndexKernel.cu:92: operator(): block: [101,0,0], thread: [63,0,0] Assertion -sizes[i] <= index && index < sizes[i] && "index out of bounds" failed.
print(torch.cuda.is_available()) → True
print(torch.version.cuda)" → 12.1