Add gpu custom op docker file.
PiperOrigin-RevId: 238549364
This commit is contained in:
parent
2fa3d94487
commit
39d4f1e1e3
21
tensorflow/tools/ci_build/Dockerfile.custom_op_gpu
Normal file
21
tensorflow/tools/ci_build/Dockerfile.custom_op_gpu
Normal 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
|
Loading…
Reference in New Issue
Block a user