import tensorflow causes kernel crash

[zeeros]

When importing the tensorflow module the following dialog appears

An unexpected error has occurred
Browser: Mozilla/5.0 (X11; Ubuntu; Linux x86_64; rv:77.0) Gecko/20100101 Firefox/77.0

Type: JS

Message: Error: Canceled future for execute_request message before replies were done

Details: Please see the console for more information

On the browser console I see this

Kernel: connected (a72fce90-4d70-47d3-8fe1-bf0d33aaf164) default.js:1487:17
Kernel: connected (4d5f461a-4dde-4d37-908c-54aa1b2415e3) default.js:1487:17
Kernel: restarting (a72fce90-4d70-47d3-8fe1-bf0d33aaf164) default.js:1487:17
Error: Canceled future for execute_request message before replies were done 2 future.js:142:30
Kernel: starting (a72fce90-4d70-47d3-8fe1-bf0d33aaf164)

The issue doesn't seems to be related to the browser since the same happens when running on Chromium

An unexpected error has occurred
Browser: Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) snap Chromium/83.0.4103.106 Chrome/83.0.4103.106 Safari/537.36

Type: JS

Message: Error: Canceled future for execute_request message before replies were done

Details: Please see the console for more information

I'm running a notebook server with an image built on a Dockerfile based on Dockerfile.rok. The error happens also with the original starting image, gcr.io/kubeflow-images-public/tensorflow-1.14.0-notebook-cpu:v0.7.0

FROM gcr.io/kubeflow-images-public/tensorflow-2.1.0-notebook-cpu:1.0.0
USER root

# Install basic dependencies
RUN apt-get update && \
    apt-get install -y --no-install-recommends \
        ca-certificates bash-completion tar less \
        python-pip python-setuptools build-essential python-dev \
        python3-pip python3-wheel && \
    rm -rf /var/lib/apt/lists/*

ENV SHELL /bin/bash
COPY bashrc /etc/bash.bashrc
RUN echo "set background=dark" >> /etc/vim/vimrc.local

# Install latest KFP SDK & Kale & JupyterLab Extension
RUN pip3 install --upgrade pip && \
    pip3 install --upgrade "jupyterlab<2.0.0" && \
    pip3 install https://storage.googleapis.com/ml-pipeline/release/latest/kfp.tar.gz --upgrade && \
    pip3 install -U kubeflow-kale && \
    jupyter labextension install kubeflow-kale-labextension

RUN echo "jovyan ALL=(ALL:ALL) NOPASSWD:ALL" > /etc/sudoers.d/jovyan
WORKDIR /home/jovyan
USER jovyan

CMD ["sh", "-c", \
     "jupyter lab --notebook-dir=/home/jovyan --ip=0.0.0.0 --no-browser \
      --allow-root --port=8888 --LabApp.token='' --LabApp.password='' \
      --LabApp.allow_origin='*' --LabApp.base_url=${NB_PREFIX}"]

All the other imports seem to work