Segmentation fault (core dumped) when training

zju3dv / PGSR

code for "PGSR: Planar-based Gaussian Splatting for Efficient and High-Fidelity Surface Reconstruction"

https://zju3dv.github.io/pgsr/

Other

403 stars 22 forks source link

Closed guishe666 closed 1 month ago

guishe666 commented 1 month ago

Hi, Thanks for the awesome work! I met a problem when training the model.

It will report "Segmentation fault (core dumped)" when calculating lncc in file loss_utils.py. It occurs when using the function F.conv2d.

I found it probably because the version of PyTorch. Related: Segmentation Fault in PyTorch 2.2 with F.conv3d function I solved by reducing the version of PyTorch to 2.0.0.

my environment info: OS: Ubuntu 22.04 Device: Nvidia RTX 4090 Cuda version: 11.8

danpeng2 commented 1 month ago

Thank you for your feedback. We have used the PyTorch 2.3.1 version and have not encountered this error.

David-Zhao-1997 commented 1 month ago

Encountered the same problem with PyTorch 2.2.1+cu118, upgrading PyTorch 2.4.0 solved the problem.