Unverified Commit 0c4e84bb authored by Yuge Zhang's avatar Yuge Zhang Committed by GitHub
Browse files

Upgrade Dockerfile to Ubuntu 18.04 (#2783)

* Upgrade dockerfile to ubuntu 18.04

* Upgrade to cuda 9.2

* Set frontend noninteractive

* Upgrade tensorflow to 1.15.2

* Fix pip version
parent e9f3cddf
# Copyright (c) Microsoft Corporation. # Copyright (c) Microsoft Corporation.
# Licensed under the MIT license. # Licensed under the MIT license.
FROM nvidia/cuda:9.0-cudnn7-runtime-ubuntu16.04 FROM nvidia/cuda:9.2-cudnn7-runtime-ubuntu18.04
LABEL maintainer='Microsoft NNI Team<nni@microsoft.com>' LABEL maintainer='Microsoft NNI Team<nni@microsoft.com>'
RUN DEBIAN_FRONTEND=noninteractive && \ ENV DEBIAN_FRONTEND=noninteractive
apt-get -y update && \
RUN apt-get -y update && \
apt-get -y install sudo \ apt-get -y install sudo \
apt-utils \ apt-utils \
git \ git \
...@@ -21,7 +22,7 @@ RUN DEBIAN_FRONTEND=noninteractive && \ ...@@ -21,7 +22,7 @@ RUN DEBIAN_FRONTEND=noninteractive && \
openssh-client \ openssh-client \
openssh-server \ openssh-server \
lsof \ lsof \
python3.5 \ python3.6 \
python3-dev \ python3-dev \
python3-pip \ python3-pip \
python3-tk \ python3-tk \
...@@ -37,7 +38,7 @@ RUN cp /usr/bin/python3 /usr/bin/python ...@@ -37,7 +38,7 @@ RUN cp /usr/bin/python3 /usr/bin/python
# #
# update pip # update pip
# #
RUN python3 -m pip install --upgrade pip setuptools==39.1.0 RUN python3 -m pip install --upgrade pip==20.0.2 setuptools==39.1.0
# numpy 1.14.3 scipy 1.1.0 # numpy 1.14.3 scipy 1.1.0
RUN python3 -m pip --no-cache-dir install \ RUN python3 -m pip --no-cache-dir install \
...@@ -46,7 +47,7 @@ RUN python3 -m pip --no-cache-dir install \ ...@@ -46,7 +47,7 @@ RUN python3 -m pip --no-cache-dir install \
# #
# Tensorflow 1.15 # Tensorflow 1.15
# #
RUN python3 -m pip --no-cache-dir install tensorflow-gpu==1.15 RUN python3 -m pip --no-cache-dir install tensorflow-gpu==1.15.0
# #
# Keras 2.1.6 # Keras 2.1.6
......
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