Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "99ebfebad1b2eb22e710a97ab9ef5bd2b6bb443e"
Unverified Commit c13392ab authored by liuzhe-lz's avatar liuzhe-lz Committed by GitHub
Browse files

Update Dockerfile (#4576)

parent 632df3ea
Hide whitespace changes
Inline Side-by-side
Showing with 26 additions and 62 deletions
Dockerfile
View file @ c13392ab
# Copyright (c) Microsoft Corporation. # Copyright (c) Microsoft Corporation.
# Licensed under the MIT license. # Licensed under the MIT license.
FROM nvidia/cuda:10.2-cudnn8-runtime-ubuntu18.04 FROM nvidia/cuda:11.3.1-cudnn8-runtime-ubuntu20.04
ARG NNI_RELEASE ARG NNI_RELEASE
...@@ -11,84 +11,48 @@ ENV DEBIAN_FRONTEND=noninteractive ...@@ -11,84 +11,48 @@ ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get -y update RUN apt-get -y update
RUN apt-get -y install \ RUN apt-get -y install \
sudo \ automake \
apt-utils \
git \
curl \
vim \
unzip \
wget \
build-essential \ build-essential \
cmake \ cmake \
libopenblas-dev \ curl \
automake \ git \
openssh-client \
openssh-server \ openssh-server \
lsof \ python3 \
python3.6 \
python3-dev \ python3-dev \
python3-pip \ python3-pip \
python3-tk \ sudo \
libcupti-dev unzip \
wget \
zip
RUN apt-get clean RUN apt-get clean
RUN rm -rf /var/lib/apt/lists/* RUN rm -rf /var/lib/apt/lists/*
#
# generate python script
#
RUN ln -s python3 /usr/bin/python RUN ln -s python3 /usr/bin/python
# RUN python3 -m pip --no-cache-dir install pip==22.0.3 setuptools==60.9.1 wheel==0.37.1
# update pip
#
RUN python3 -m pip install --upgrade pip==20.2.4 setuptools==50.3.2
# numpy 1.19.5 scipy 1.5.4 RUN python3 -m pip --no-cache-dir install \
RUN python3 -m pip --no-cache-dir install numpy==1.19.5 scipy==1.5.4 lightgbm==3.3.2 \
numpy==1.22.2 \
pandas==1.4.1 \
scikit-learn==1.0.2 \
scipy==1.8.0
# RUN python3 -m pip --no-cache-dir install \
# TensorFlow torch==1.10.2+cu113 \
# torchvision==0.11.3+cu113 \
RUN python3 -m pip --no-cache-dir install tensorflow==2.3.1 torchaudio==0.10.2+cu113 \
-f https://download.pytorch.org/whl/cu113/torch_stable.html
RUN python3 -m pip --no-cache-dir install pytorch-lightning==1.5.10
# RUN python3 -m pip --no-cache-dir install tensorflow==2.8.0
# Keras
#
RUN python3 -m pip --no-cache-dir install Keras==2.4.3
# RUN python3 -m pip --no-cache-dir install azureml==0.2.7 azureml-sdk==1.38.0
# PyTorch
#
RUN python3 -m pip --no-cache-dir install torch==1.7.1 torchvision==0.8.2 pytorch-lightning==1.3.3
#
# sklearn 0.24.1
#
RUN python3 -m pip --no-cache-dir install scikit-learn==0.24.1
#
# pandas==0.23.4 lightgbm==2.2.2
#
RUN python3 -m pip --no-cache-dir install pandas==1.1 lightgbm==2.2.2
#
# Install NNI
#
COPY dist/nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl . COPY dist/nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl .
RUN python3 -m pip install nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl RUN python3 -m pip install nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl
RUN rm nni-${NNI_RELEASE}-py3-none-manylinux1_x86_64.whl
# ENV PATH=/root/.local/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/bin:/usr/bin:/usr/sbin
# Vision patch. Need del later
#
COPY test/vso_tools/interim_patch.py .
RUN python3 interim_patch.py
#
# install aml package
#
RUN python3 -m pip --no-cache-dir install azureml
RUN python3 -m pip --no-cache-dir install azureml-sdk
ENV PATH=/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/root/.local/bin:/usr/bin:/bin:/sbin
WORKDIR /root WORKDIR /root
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