This sounds as if the JIT would kick in and compile the CUDA kernels for your architecture, which seem to be missing from the used binary.
How did you install/build PyTorch and which GPU are you using?


Once 1. is solved, this might also be fixed.