Add gpu custom op docker file.

PiperOrigin-RevId: 238549364
This commit is contained in:
Yifei Feng 2019-03-14 17:02:33 -07:00 committed by TensorFlower Gardener
parent 2fa3d94487
commit 39d4f1e1e3

View File

@ -0,0 +1,21 @@
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu14.04
LABEL maintainer="Yifei Feng <yifeif@google.com>"
# Copy and run the install scripts.
COPY install/*.sh /install/
RUN /install/install_bootstrap_deb_packages.sh
RUN add-apt-repository -y ppa:openjdk-r/ppa && \
add-apt-repository -y ppa:george-edison55/cmake-3.x
RUN /install/install_deb_packages.sh
RUN /install/install_pip_packages.sh
RUN /install/install_bazel.sh
RUN /install/install_proto3.sh
RUN /install/install_buildifier.sh
RUN /install/install_auditwheel.sh
RUN /install/install_golang.sh
# Set up the master bazelrc configuration file.
COPY install/.bazelrc /etc/bazel.bazelrc
ENV TF_NEED_CUDA 1