experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Automated rollback of commit 220e8e059a

Browse Source 
PiperOrigin-RevId: 249052565
This commit is contained in:
Guangda Lai 2019-05-20 08:22:44 -07:00 committed by TensorFlower Gardener
parent ba6a8875dc
commit 935ac2b733
21 changed files with 71 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/tools/ci_build/Dockerfile.gpu
View File

@ -8,8 +8,10 @@ RUN cp -P /usr/include/cudnn.h /usr/local/cuda/include
RUN cp -P /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64 RUN cp -P /usr/lib/x86_64-linux-gnu/libcudnn* /usr/local/cuda/lib64
# Installs TensorRT, which is not included in NVIDIA Docker containers. # Installs TensorRT, which is not included in NVIDIA Docker containers.
RUN apt-get update && apt-get install -y --no-install-recommends \ RUN apt-get update \
libnvinfer5=5.1.5-1+cuda10.0 libnvinfer-dev=5.1.5-1+cuda10.0 && apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda10.0 \
&& apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda10.0 libnvinfer-dev=5.0.2-1+cuda10.0
# Copy and run the install scripts. # Copy and run the install scripts.
COPY install/*.sh /install/ COPY install/*.sh /install/

4
tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu14.04
View File

@ -19,7 +19,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends ca-certificates
ENV CUDA_VERSION 10.0.130 ENV CUDA_VERSION 10.0.130
ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1 ENV CUDA_PKG_VERSION 10-0=$CUDA_VERSION-1
ENV CUDNN_VERSION 7.3.1.20 ENV CUDNN_VERSION 7.3.1.20
ENV TENSORRT_VERSION 5.1.5 ENV TENSORRT_VERSION 5.0.2
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0,driver>=410" ENV NVIDIA_REQUIRE_CUDA "cuda>=10.0,driver>=410"
ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_VISIBLE_DEVICES all
@ -47,6 +47,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
libcudnn7=$CUDNN_VERSION-1+cuda10.0 \ libcudnn7=$CUDNN_VERSION-1+cuda10.0 \
libcudnn7=$CUDNN_VERSION-1+cuda10.0 \ libcudnn7=$CUDNN_VERSION-1+cuda10.0 \
libcudnn7-dev=$CUDNN_VERSION-1+cuda10.0 \ libcudnn7-dev=$CUDNN_VERSION-1+cuda10.0 \
nvinfer-runtime-trt-repo-ubuntu1604-$TENSORRT_VERSION-ga-cuda10.0 && \
apt-get update && apt-get install -y --no-install-recommends \
libnvinfer5=$TENSORRT_VERSION-1+cuda10.0 \ libnvinfer5=$TENSORRT_VERSION-1+cuda10.0 \
libnvinfer-dev=$TENSORRT_VERSION-1+cuda10.0 && \ libnvinfer-dev=$TENSORRT_VERSION-1+cuda10.0 && \
ln -s cuda-10.0 /usr/local/cuda && \ ln -s cuda-10.0 /usr/local/cuda && \

4
tensorflow/tools/ci_build/Dockerfile.rbe.cuda9.0-cudnn7-ubuntu14.04
View File

@ -25,7 +25,7 @@ ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH}
ENV NVIDIA_VISIBLE_DEVICES all ENV NVIDIA_VISIBLE_DEVICES all
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV NVIDIA_DRIVER_CAPABILITIES compute,utility
ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0" ENV NVIDIA_REQUIRE_CUDA "cuda>=9.0"
ENV TENSORRT_VERSION 5.1.5 ENV TENSORRT_VERSION 5.0.2
ENV CUDNN_VERSION 7.1.4.18 ENV CUDNN_VERSION 7.1.4.18
# TODO(b/110903506): /usr/loca/cuda/lib64/stubs should not be needed in # TODO(b/110903506): /usr/loca/cuda/lib64/stubs should not be needed in
@ -52,6 +52,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
cuda-cublas-dev-9-0=9.0.176.4-1 \ cuda-cublas-dev-9-0=9.0.176.4-1 \
libcudnn7-dev=$CUDNN_VERSION-1+cuda9.0 \ libcudnn7-dev=$CUDNN_VERSION-1+cuda9.0 \
libcudnn7=$CUDNN_VERSION-1+cuda9.0 \ libcudnn7=$CUDNN_VERSION-1+cuda9.0 \
nvinfer-runtime-trt-repo-ubuntu1604-$TENSORRT_VERSION-ga-cuda9.0 && \
apt-get update && apt-get install -y --no-install-recommends \
libnvinfer5=$TENSORRT_VERSION-1+cuda9.0 \ libnvinfer5=$TENSORRT_VERSION-1+cuda9.0 \
libnvinfer-dev=$TENSORRT_VERSION-1+cuda9.0 && \ libnvinfer-dev=$TENSORRT_VERSION-1+cuda9.0 && \
ln -s cuda-9.0 /usr/local/cuda && \ ln -s cuda-9.0 /usr/local/cuda && \

2
tensorflow/tools/dockerfiles/dockerfiles/cpu-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} as base FROM ubuntu:${UBUNTU_VERSION} as base

2
tensorflow/tools/dockerfiles/dockerfiles/cpu.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} as base FROM ubuntu:${UBUNTU_VERSION} as base

2
tensorflow/tools/dockerfiles/dockerfiles/devel-cpu-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} AS base FROM ubuntu:${UBUNTU_VERSION} AS base

2
tensorflow/tools/dockerfiles/dockerfiles/devel-cpu.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} AS base FROM ubuntu:${UBUNTU_VERSION} AS base

10
tensorflow/tools/dockerfiles/dockerfiles/devel-gpu-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -61,10 +61,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a
RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update \ RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
&& apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
libnvinfer5=5.1.5-1+cuda${CUDA} \ libnvinfer5=5.0.2-1+cuda${CUDA} \
libnvinfer-dev=5.1.5-1+cuda${CUDA} \ libnvinfer-dev=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*; } && rm -rf /var/lib/apt/lists/*; }

10
tensorflow/tools/dockerfiles/dockerfiles/devel-gpu.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -61,10 +61,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a
RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update \ RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
&& apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
libnvinfer5=5.1.5-1+cuda${CUDA} \ libnvinfer5=5.0.2-1+cuda${CUDA} \
libnvinfer-dev=5.1.5-1+cuda${CUDA} \ libnvinfer-dev=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*; } && rm -rf /var/lib/apt/lists/*; }

9
tensorflow/tools/dockerfiles/dockerfiles/gpu-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -50,9 +50,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
unzip unzip
RUN [ ${ARCH} = ppc64le ] || (apt-get update \ RUN [ ${ARCH} = ppc64le ] || (apt-get update && \
&& apt-get install -y --no-install-recommends \ apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
libnvinfer5=5.1.5-1+cuda${CUDA} \ && apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*) && rm -rf /var/lib/apt/lists/*)

9
tensorflow/tools/dockerfiles/dockerfiles/gpu.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -50,9 +50,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
unzip unzip
RUN [ ${ARCH} = ppc64le ] || (apt-get update \ RUN [ ${ARCH} = ppc64le ] || (apt-get update && \
&& apt-get install -y --no-install-recommends \ apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
libnvinfer5=5.1.5-1+cuda${CUDA} \ && apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*) && rm -rf /var/lib/apt/lists/*)

2
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/cpu-ppc64le-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} as base FROM ubuntu:${UBUNTU_VERSION} as base

2
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/cpu-ppc64le.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} as base FROM ubuntu:${UBUNTU_VERSION} as base

2
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/devel-cpu-ppc64le-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} AS base FROM ubuntu:${UBUNTU_VERSION} AS base

2
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/devel-cpu-ppc64le.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
FROM ubuntu:${UBUNTU_VERSION} AS base FROM ubuntu:${UBUNTU_VERSION} AS base

10
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/devel-gpu-ppc64le-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -61,10 +61,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a
RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update \ RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
&& apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
libnvinfer5=5.1.5-1+cuda${CUDA} \ libnvinfer5=5.0.2-1+cuda${CUDA} \
libnvinfer-dev=5.1.5-1+cuda${CUDA} \ libnvinfer-dev=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*; } && rm -rf /var/lib/apt/lists/*; }

10
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/devel-gpu-ppc64le.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -61,10 +61,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a
RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update \ RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
&& apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
libnvinfer5=5.1.5-1+cuda${CUDA} \ libnvinfer5=5.0.2-1+cuda${CUDA} \
libnvinfer-dev=5.1.5-1+cuda${CUDA} \ libnvinfer-dev=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*; } && rm -rf /var/lib/apt/lists/*; }

9
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/gpu-ppc64le-jupyter.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -50,9 +50,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
unzip unzip
RUN [ ${ARCH} = ppc64le ] || (apt-get update \ RUN [ ${ARCH} = ppc64le ] || (apt-get update && \
&& apt-get install -y --no-install-recommends \ apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
libnvinfer5=5.1.5-1+cuda${CUDA} \ && apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*) && rm -rf /var/lib/apt/lists/*)

9
tensorflow/tools/dockerfiles/dockerfiles/ppc64le/gpu-ppc64le.Dockerfile
View File

@ -19,7 +19,7 @@
# throughout. Please refer to the TensorFlow dockerfiles documentation # throughout. Please refer to the TensorFlow dockerfiles documentation
# for more information. # for more information.
ARG UBUNTU_VERSION=18.04 ARG UBUNTU_VERSION=16.04
ARG ARCH= ARG ARCH=
ARG CUDA=10.0 ARG CUDA=10.0
@ -50,9 +50,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
unzip unzip
RUN [ ${ARCH} = ppc64le ] || (apt-get update \ RUN [ ${ARCH} = ppc64le ] || (apt-get update && \
&& apt-get install -y --no-install-recommends \ apt-get install nvinfer-runtime-trt-repo-ubuntu1604-5.0.2-ga-cuda${CUDA} \
libnvinfer5=5.1.5-1+cuda${CUDA} \ && apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*) && rm -rf /var/lib/apt/lists/*)

8
tensorflow/tools/dockerfiles/partials/ubuntu/devel-nvidia.partial.Dockerfile
View File

@ -38,10 +38,12 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \ find /usr/local/cuda-${CUDA}/lib64/ -type f -name 'lib*_static.a' -not -name 'libcudart_static.a' -delete && \
rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a rm /usr/lib/${LIB_DIR_PREFIX}-linux-gnu/libcudnn_static_v7.a
RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update \ RUN [[ "${ARCH}" = "ppc64le" ]] || { apt-get update && \
apt-get install nvinfer-runtime-trt-repo-ubuntu1804-5.0.2-ga-cuda${CUDA} \
&& apt-get update \
&& apt-get install -y --no-install-recommends \ && apt-get install -y --no-install-recommends \
libnvinfer5=5.1.5-1+cuda${CUDA} \ libnvinfer5=5.0.2-1+cuda${CUDA} \
libnvinfer-dev=5.1.5-1+cuda${CUDA} \ libnvinfer-dev=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*; } && rm -rf /var/lib/apt/lists/*; }

7
tensorflow/tools/dockerfiles/partials/ubuntu/nvidia.partial.Dockerfile
View File

@ -27,9 +27,10 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
software-properties-common \ software-properties-common \
unzip unzip
RUN [ ${ARCH} = ppc64le ] || (apt-get update \ RUN [ ${ARCH} = ppc64le ] || (apt-get update && \
&& apt-get install -y --no-install-recommends \ apt-get install nvinfer-runtime-trt-repo-ubuntu1804-5.0.2-ga-cuda${CUDA} \
libnvinfer5=5.1.5-1+cuda${CUDA} \ && apt-get update \
&& apt-get install -y --no-install-recommends libnvinfer5=5.0.2-1+cuda${CUDA} \
&& apt-get clean \ && apt-get clean \
&& rm -rf /var/lib/apt/lists/*) && rm -rf /var/lib/apt/lists/*)