Segmentation fault (core dumped)

[Goddard-Wen]

(foundpose_gpu) goddard@GoddardWen:/mnt/d/000Project/foundpose$ python scripts/gen_repre.py --opts-path configs/gen_repre/lmo.json
I0309 17:36:05.598 13248 loader.py:74] Loading faiss with AVX512 support.
I0309 17:36:05.628 13248 loader.py:76] Successfully loaded faiss with AVX512 support.
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/swiglu_ffn.py:43: UserWarning: xFormers is available (SwiGLU)
warnings.warn("xFormers is available (SwiGLU)")
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/attention.py:27: UserWarning: xFormers is available (Attention)
warnings.warn("xFormers is available (Attention)")
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/block.py:33: UserWarning: xFormers is available (Block)
warnings.warn("xFormers is available (Block)")
I0309 17:36:07.969 13248 vision_transformer.py:122] using MLP layer as FFN
Device: cuda
I0309 17:36:08.548 13248 misc.py:41] Time for preparation: 0.18010s
I0309 17:36:08.658 13248 gen_repre.py:108] Processing dataset 0/798,
Segmentation fault (core dumped)

[Goddard-Wen]

How to solve this problem, I have been dealing with it all day and don't know what to do.

[Goddard-Wen]

After locating, I confirmed that the problem occurred at line 168, but I still don't know how to solve it

(foundpose_gpu) goddard@GoddardWen:/mnt/d/000Project/freepose/foundpose$ python -X faulthandler scripts/gen_repr
e.py --opts-path configs/gen_repre/lmo.json
I0309 18:05:40.404 15064 loader.py:74] Loading faiss with AVX512 support.
I0309 18:05:40.466 15064 loader.py:76] Successfully loaded faiss with AVX512 support.
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/swiglu_ffn.py:43: UserWarning: xFormers is available (SwiGLU)
warnings.warn("xFormers is available (SwiGLU)")
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/attention.py:27: UserWarning: xFormers is available (Attention)
warnings.warn("xFormers is available (Attention)")
/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/block.py:33: UserWarning: xFormers is available (Block)
warnings.warn("xFormers is available (Block)")
I0309 18:05:43.3 15064 vision_transformer.py:122] using MLP layer as FFN
Device: cuda
I0309 18:05:43.605 15064 misc.py:41] Time for preparation: 0.19474s
I0309 18:05:43.723 15064 gen_repre.py:108] Processing dataset 0/798,
Fatal Python error: Segmentation fault

Current thread 0x00007fc46afe3740 (most recent call first):
File "/home/goddard/anaconda3/envs/foundpose_gpu/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 456 in _conv_forward
File "/home/goddard/anaconda3/envs/foundpose_gpu/lib/python3.9/site-packages/torch/nn/modules/conv.py", line 460 in forward
File "/home/goddard/anaconda3/envs/foundpose_gpu/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1541 in _call_impl
File "/home/goddard/anaconda3/envs/foundpose_gpu/lib/python3.9/site-packages/torch/nn/modules/module.py", line 1532 in _wrapped_call_impl
File "/mnt/d/000Project/foundpose/external/dinov2/dinov2/layers/patch_embed.py", line 75 in forward
File "/mnt/d/000Project/freepose/foundpose/scripts/gen_repre.py", line 256 in generate_repre
File "/mnt/d/000Project/freepose/foundpose/scripts/gen_repre.py", line 397 in generate_repre_from_list
File "/mnt/d/000Project/freepose/foundpose/scripts/gen_repre.py", line 401 in main
File "/mnt/d/000Project/freepose/foundpose/scripts/gen_repre.py", line 407 in
Segmentation fault (core dumped)