Segmentation fault when run it in server

RLlib
1

How severe does this issue affect your experience of using Ray?

  • High: It blocks me to complete my task.

Hello, I am trying to use impala algorithm in my server. When I run my code locally, it works well. However, when I run the code at my server by slrum, error below appears.

/var/spool/slurmd/job4781601/slurm_script: line 19: 48241 Segmentation fault      (core dumped) python launch.py

This error appears when I just import ray libraries.

import ray
from ray import air, tune
from ray.rllib.examples.env.look_and_push import LookAndPush, OneHot
from ray.rllib.examples.env.repeat_after_me_env import RepeatAfterMeEnv
from ray.rllib.examples.env.repeat_initial_obs_env import RepeatInitialObsEnv
from ray.rllib.examples.env.stateless_cartpole import StatelessCartPole
from ray.rllib.utils.framework import try_import_tf
from ray.rllib.utils.test_utils import check_learning_achieved
from ray.tune import registry
from ray.tune.logger import pretty_print

I think this is because my code use too much resources in the server but It uses so small resources that I can run it in my laptop.
I am suffering this problem for a week… Is there any solution? thank you.