segment fault when computing loss

[NIRVANALAN]

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]

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]

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]

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