Skip to content

Commit

Permalink
Upgrade Dockerfile to Ubuntu 18.04 (microsoft#2783)
Browse files Browse the repository at this point in the history
* Upgrade dockerfile to ubuntu 18.04

* Upgrade to cuda 9.2

* Set frontend noninteractive

* Upgrade tensorflow to 1.15.2

* Fix pip version
  • Loading branch information
ultmaster authored Aug 12, 2020
1 parent e9f3cdd commit 0c4e84b
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions deployment/docker/Dockerfile
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
# Copyright (c) Microsoft Corporation.
# 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<[email protected]>'

RUN DEBIAN_FRONTEND=noninteractive && \
apt-get -y update && \
ENV DEBIAN_FRONTEND=noninteractive

RUN apt-get -y update && \
apt-get -y install sudo \
apt-utils \
git \
Expand All @@ -21,7 +22,7 @@ RUN DEBIAN_FRONTEND=noninteractive && \
openssh-client \
openssh-server \
lsof \
python3.5 \
python3.6 \
python3-dev \
python3-pip \
python3-tk \
Expand All @@ -37,7 +38,7 @@ RUN cp /usr/bin/python3 /usr/bin/python
#
# 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
RUN python3 -m pip --no-cache-dir install \
Expand All @@ -46,7 +47,7 @@ RUN python3 -m pip --no-cache-dir install \
#
# 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
Expand Down

0 comments on commit 0c4e84b

Please sign in to comment.