Loading extension module _prroi_pooling Segmentation fault

Sometimes an error occurs when running object tracking on Jetson XAVIER.
It happens after the loading of the extension prroi_pooling.

Detected CUDA files, patching ldflags
Emitting ninja build file /home/roboneers/.cache/torch_extensions/_prroi_pooling/build.ninja...
Building extension module _prroi_pooling...
Allowing ninja to set a default number of workers... (overridable by setting the environment variable MAX_JOBS=N)
ninja: no work to do.
Loading extension module _prroi_pooling...
Segmentation fault

Maybe someone has encountered this problem and knows how to solve it.

Python: 3.6
Torch: 1.8.0
Numpy: 1.18.5

Could you update to the latest PyTorch release (1.9.0) and check, if you are still seeing this issue?
If so, could you post an executable code snippet to reproduce this error?