The command should install PyTorch with CUDA10.1, while apparently you are using 9.2?
Could you explain your use case a bit?
E.g. what kind of data are you using (which shape)? Does it have variable shapes?
Do you see an increased memory usage on your GPU using nvidia-smi or does it stay constant during training? What kind of model are you using?
If possible, could you try to run your code with random inputs and try to reproduce this error with them? If you see the same error, could you post a small code snippet to reproduce this issue?