STT-tensorflow/tensorflow/tools/dockerfiles/partials/ubuntu/devel-cpu.partial.Dockerfile
Austin Anderson 57826bf39d Fix images and tests for Cuda 10.1 and TensorRT 6
This change makes the dockerfile build tests pass. They have been broken
ever since CUDA 10.1, and I fixed them by removing dependency on the old
ci_build configure scripts and by updating TensorRT to version 6.

Unfortunately, TensorRT builds seem to be broken at head (see
https://github.com/tensorflow/tensorflow/issues/35115), so I've had to
disable TensorRT for the build tests. However, this should make the
devel images able to build regularly again.
2019-12-19 12:44:39 -08:00

33 lines
1.0 KiB
Docker

FROM ubuntu:${UBUNTU_VERSION} AS base
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential \
curl \
git \
libcurl3-dev \
libfreetype6-dev \
libhdf5-serial-dev \
libzmq3-dev \
pkg-config \
rsync \
software-properties-common \
sudo \
unzip \
zip \
zlib1g-dev \
openjdk-8-jdk \
openjdk-8-jre-headless \
&& \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
ENV CI_BUILD_PYTHON python
# CACHE_STOP is used to rerun future commands, otherwise cloning tensorflow will be cached and will not pull the most recent version
ARG CACHE_STOP=1
# Check out TensorFlow source code if --build-arg CHECKOUT_TF_SRC=1
ARG CHECKOUT_TF_SRC=0
# In case of Python 2.7+ we need to add passwd entries for user and group id
RUN chmod a+w /etc/passwd /etc/group
RUN test "${CHECKOUT_TF_SRC}" -eq 1 && git clone https://github.com/tensorflow/tensorflow.git /tensorflow_src || true