22 lines
661 B
Docker
22 lines
661 B
Docker
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
|