The cuda version mistmatch might be causing this, since PyTorch seems to be using 10.1 but your system seems to have 11.4 installed. Could you try to install Cuda toolkit 10.1 and see if that fixes the issue?