Segmentation fault is occar

[Sueoka-ppc]

When I run jukebox sample code, Segmentation fault is happend

code is blow

import jukebox
import torch as t
import librosa
import os
#from IPython.display import Audio
from jukebox.make_models import make_vqvae, make_prior, MODELS, make_model
from jukebox.hparams import Hyperparams, setup_hparams
from jukebox.sample import sample_single_window, _sample, \
                           sample_partial_window, upsample
from jukebox.utils.dist_utils import setup_dist_from_mpi
from jukebox.utils.torch_utils import empty_cache
rank, local_rank, device = setup_dist_from_mpi()      <== when this code run, cash is happend

Python interprinter logs is blow

Python 3.7.5 (default, Oct 25 2019, 16:29:01) 
[GCC 7.3.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import jukebox
>>> import torch as t
>>> import librosa
>>> import os
>>> from jukebox.make_models import make_vqvae, make_prior, MODELS, make_model
>>> from jukebox.hparams import Hyperparams, setup_hparams
>>> from jukebox.sample import sample_single_window, _sample, \
...                            sample_partial_window, upsample
>>> from jukebox.utils.dist_utils import setup_dist_from_mpi
>>> from jukebox.utils.torch_utils import empty_cache
>>> rank, local_rank, device = setup_dist_from_mpi()
Segmentation fault (core dump)

conda info

     active environment : jukebox
    active env location : /home/sueoka/anaconda3_2/envs/jukebox
            shell level : 2
       user config file : /home/sueoka/.condarc
 populated config files : /home/sueoka/.condarc
          conda version : 4.10.1
    conda-build version : 3.21.4
         python version : 3.8.8.final.0
       virtual packages : __cuda=11.1=0
                          __linux=4.15.0=0
                          __glibc=2.27=0
                          __unix=0=0
                          __archspec=1=ppc64le
       base environment : /home/sueoka/anaconda3_2  (writable)
      conda av data dir : /home/sueoka/anaconda3_2/etc/conda
  conda av metadata url : https://repo.anaconda.com/pkgs/main
           channel URLs : https://opence.mit.edu/linux-ppc64le
                          https://opence.mit.edu/noarch
                          https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda/linux-ppc64le
                          https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda/noarch
                          https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda-early-access/linux-ppc64le
                          https://public.dhe.ibm.com/ibmdl/export/pub/software/server/ibm-ai/conda-early-access/noarch
                          https://repo.anaconda.com/pkgs/main/linux-ppc64le
                          https://repo.anaconda.com/pkgs/main/noarch
                          https://repo.anaconda.com/pkgs/r/linux-ppc64le
                          https://repo.anaconda.com/pkgs/r/noarch
          package cache : /home/sueoka/anaconda3_2/pkgs
                          /home/sueoka/.conda/pkgs
       envs directories : /home/sueoka/anaconda3_2/envs
                          /home/sueoka/.conda/envs
               platform : linux-ppc64le
             user-agent : conda/4.10.1 requests/2.25.1 CPython/3.8.8 Linux/4.15.0-112-generic ubuntu/18.04.5 glibc/2.27
                UID:GID : 1000:1000
             netrc file : None
           offline mode : False