The code breaks at line 139 of my python file a.py where I am loading my model on cuda using model.cuda().
This problem doesn’t happen when I use Cuda 11.3 and Pytorch 1.9.
May I know how can I use Cuda 10.1 on RTX 3090 because other libraries have dependency and I need to use cuda 10.1.