CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start

[Dwight_Foster]

Is your pytorch version compatible with cuda 11.2