segment fault when computing loss

GAP-LAB-CUHK-SZ / Total3DUnderstanding

Implementation of CVPR'20 Oral: Total3DUnderstanding: Joint Layout, Object Pose and Mesh Reconstruction for Indoor Scenes from a Single Image

MIT License

415 stars 50 forks source link

segment fault when computing loss #12

Open NIRVANALAN opened 3 years ago

NIRVANALAN commented 3 years ago

Hi, I create the env using conda and right torch/cuda version. But when I try to run any script, it always produces Segmentation fault. I have tested my env on other tasks like minist conv classifications. I am on Centos, and I have tried gcc 4.8, 7.3 and 8.2. Any suggestions?

yinyunie commented 3 years ago

Hi,

Its strange. The code is developed under Ubuntu but I think it is the same for Centos. Can you post any debug info?

WenM1222 commented 3 years ago

Hi, I also got segmentation fault error while I ran the code inside docker container. But no problem in host PC. The system is Ubuntu18.04.

alando46 commented 3 years ago

Unpinning - cudatoolkit=9.0 to - cudatoolkit and let conda find the best version worked for me.