Skip to content

GitLab

Unverified Commit 3e935d9d authored by Guolin Ke's avatar Guolin Ke Committed by GitHub
Browse files

Update Dockerfile

parent 552e8b25
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 11 deletions
docker/Dockerfile
View file @ 3e935d9d
# ==================================================================
# module list
# ------------------------------------------------------------------
# python 3.8 (conda)
# pytorch 1.11.0 (conda)
# apex from github
# ==================================================================
FROM nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04 FROM nvidia/cuda:11.3.0-cudnn8-devel-ubuntu20.04
ENV LANG C.UTF-8 ENV LANG C.UTF-8
...@@ -60,6 +52,10 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ ...@@ -60,6 +52,10 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \
debhelper \ debhelper \
nfs-common nfs-common
# ==================================================================
# InfiniBand & RDMA
# ------------------------------------------------------------------
RUN cd /tmp && \ RUN cd /tmp && \
wget -q http://content.mellanox.com/ofed/MLNX_OFED-${OFED_VERSION}/MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu20.04-x86_64.tgz && \ wget -q http://content.mellanox.com/ofed/MLNX_OFED-${OFED_VERSION}/MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu20.04-x86_64.tgz && \
tar xzf MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu20.04-x86_64.tgz && \ tar xzf MLNX_OFED_LINUX-${OFED_VERSION}-ubuntu20.04-x86_64.tgz && \
...@@ -92,14 +88,11 @@ RUN wget -O ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest ...@@ -92,14 +88,11 @@ RUN wget -O ~/miniconda.sh https://repo.continuum.io/miniconda/Miniconda3-latest
rm ~/miniconda.sh rm ~/miniconda.sh
ENV PATH /opt/conda/bin:$PATH ENV PATH /opt/conda/bin:$PATH
RUN conda install -y python=3.8 && conda clean -ya RUN conda install -y python=3.8 && conda clean -ya
RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \ RUN conda install -y scipy scikit-learn pyyaml tensorboard tensorboardX && \
conda clean -ya conda clean -ya
# install pytorch
RUN ldconfig RUN ldconfig
# ================================================================== # ==================================================================
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment