Floating point exception (core dumped)

[diamond0910]

Thank you for your work. I create the python env using the given environment.yml. But I got the error during training. My platform is 20.04 ubuntu, and my cuda version is 11.4.

Here is my pip list.