| 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 |
