I have an RTX 3080 and I want to use it to train a deep learning model. Both CUDA and Pytorch I installed are version 10.2.  However, when I train this deep learning model, it freezes when loaded into CUDA.  Is there any solution?