Dockerfile for my issue in BytePS

Dockerfile

FROM nvidia/cuda:11.0-devel-ubuntu18.04

ARG https_proxy
ARG http_proxy

ARG BYTEPS_BASE_PATH=/usr/local
ARG BYTEPS_PATH=$BYTEPS_BASE_PATH/byteps
ARG BYTEPS_GIT_LINK=https://github.com/ruipeterpan/byteps
ARG BYTEPS_BRANCH=master
ARG TORCHELASTIC_GIT_LINK=https://github.com/ruipeterpan/elastic
ARG TORCHELASTIC_PATH=$BYTEPS_BASE_PATH/elastic

ARG DEBIAN_FRONTEND=noninteractive
RUN apt-get update
RUN apt-get install -y --allow-downgrades --allow-change-held-packages --no-install-recommends \
        build-essential \
        tzdata \
        ca-certificates \
        git \
        curl \
        wget \
        vim \
        gdb \
        cmake \
        lsb-release \
        libcudnn8=8.0.5.39-1+cuda11.0 \
        libnuma-dev \
        ibverbs-providers \
        librdmacm-dev \
        ibverbs-utils \
        rdmacm-utils \
        libibverbs-dev \
        python3.8 \
        python3.8-dev \
        python3-pip \
        python3-setuptools \
        libnccl2=2.7.8-1+cuda11.0 \
        libnccl-dev=2.7.8-1+cuda11.0

# install framework
# note: for tf <= 1.14, you need gcc-4.9
ARG FRAMEWORK=pytorch
RUN if [ "$FRAMEWORK" = "tensorflow" ]; then \
        pip3 install --upgrade pip; \
        pip3 install -U tensorflow-gpu==1.15.0; \
    elif [ "$FRAMEWORK" = "pytorch" ]; then \
        python3.8 -m pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 -f https://download.pytorch.org/whl/torch_stable.html; \
    elif [ "$FRAMEWORK" = "mxnet" ]; then \
        pip3 install -U mxnet-cu100==1.5.0; \
    else \
        echo "unknown framework: $FRAMEWORK"; \
        exit 1; \
    fi

ENV LD_LIBRARY_PATH /usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH

RUN cd $BYTEPS_BASE_PATH &&\
    git clone --recursive -b $BYTEPS_BRANCH $BYTEPS_GIT_LINK &&\
    cd $BYTEPS_PATH &&\
    python3.8 setup.py install

RUN cd $BYTEPS_BASE_PATH &&\
    git clone --recursive -b $BYTEPS_BRANCH $TORCHELASTIC_GIT_LINK &&\
    cd $TORCHELASTIC_PATH &&\
    python3.8 setup.py install