From a8e50731d345843ead6a3eda43777b0815f0cf5b Mon Sep 17 00:00:00 2001 From: Jamie Dabu Date: Wed, 23 Dec 2020 16:54:44 -0600 Subject: [PATCH] Updates Dockerfile with new repos and names for Intel DNN and tensorflow gpu nightly --- .../neural_network/Dockerfile.all-jupyter-py36-cu101 | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/python/neural_network/Dockerfile.all-jupyter-py36-cu101 b/python/neural_network/Dockerfile.all-jupyter-py36-cu101 index 4cc645d..ce3db5c 100644 --- a/python/neural_network/Dockerfile.all-jupyter-py36-cu101 +++ b/python/neural_network/Dockerfile.all-jupyter-py36-cu101 @@ -212,7 +212,7 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ # ------------------------------------------------------------------ $PIP_INSTALL \ - tf-nightly-gpu-2.0-preview \ + tf-nightly-gpu \ && \ # ================================================================== @@ -320,11 +320,11 @@ RUN APT_INSTALL="apt-get install -y --no-install-recommends" && \ ln -s /usr/lib/x86_64-linux-gnu/libmpi_cxx.so.20 /usr/lib/x86_64-linux-gnu/libmpi_cxx.so.1 && \ ln -s /usr/lib/x86_64-linux-gnu/libmpi.so.20.10.1 /usr/lib/x86_64-linux-gnu/libmpi.so.12 && \ - wget --no-verbose -O - https://github.com/01org/mkl-dnn/releases/download/v0.14/mklml_lnx_2018.0.3.20180406.tgz | tar -xzf - && \ - cp mklml*/* /usr/local -r && \ + wget --no-verbose -O - https://github.com/oneapi-src/oneDNN/releases/download/v1.4/dnnl_lnx_1.4.0_cpu_iomp.tgz | tar -xzf - && \ + cp dnnl*/* /usr/local -r && \ - wget --no-verbose -O - https://github.com/01org/mkl-dnn/archive/v0.14.tar.gz | tar -xzf - && \ - cd mkl-dnn-0.14 && mkdir build && cd build && \ + wget --no-verbose -O - https://github.com/oneapi-src/oneDNN/archive/v1.4.tar.gz | tar -xzf - && \ + cd oneDNN-1.4 && mkdir build && cd build && \ ln -s /usr/local external && \ cmake -D CMAKE_BUILD_TYPE=RELEASE \ -D CMAKE_INSTALL_PREFIX=/usr/local \