Internal change.
PiperOrigin-RevId: 253331306
This commit is contained in:
parent
51e24dacba
commit
5e545bbb5c
@ -19,15 +19,9 @@
|
||||
FROM nvidia/cuda:10.0-cudnn7-devel-centos6
|
||||
LABEL maintainer="Amit Patankar <amitpatankar@google.com>"
|
||||
|
||||
# Install packages required to build tensorflow.
|
||||
RUN yum install -y centos-release-scl && \
|
||||
yum install -y \
|
||||
devtoolset-7 \
|
||||
java-1.8.0-openjdk-devel \
|
||||
patch \
|
||||
rh-python35 \
|
||||
wget && \
|
||||
yum clean all -y
|
||||
# Install yum packages required to build tensorflow.
|
||||
COPY install/*.sh /install/
|
||||
RUN bash install/install_yum_packages.sh
|
||||
|
||||
# TODO(klimek): Once nvidia provides a way to install tensorrt from within
|
||||
# docker, switch to that. Until then, we copy the tensorrt rpm into the docker
|
||||
@ -44,9 +38,7 @@ env PATH="/opt/rh/python27/root/usr/bin:/opt/rh/devtoolset-7/root/usr/bin:/usr/l
|
||||
PERL5LIB="/opt/rh/devtoolset-7/root//usr/lib64/perl5/vendor_perl:/opt/rh/devtoolset-7/root/usr/lib/perl5:/opt/rh/devtoolset-7/root//usr/share/perl5/vendor_perl" \
|
||||
PKG_CONFIG_PATH="/opt/rh/python27/root/usr/lib64/pkgconfig/"
|
||||
|
||||
# Install pip packages needed to build tensorflow.
|
||||
COPY install/*.sh /install/
|
||||
RUN bash install/install_yum_packages.sh
|
||||
# Install python3.6 and pip packages needed to build tensorflow.
|
||||
RUN bash install/install_centos_python36.sh
|
||||
RUN bash install/install_centos_pip_packages.sh
|
||||
|
||||
|
@ -4,6 +4,6 @@ container_digests = {
|
||||
"centos6": "sha256:5450c88850260a5d46f4942c3807e2240eb5e792fce5e788bacf2317309969f1",
|
||||
"cuda10.0-cudnn7-ubuntu14.04": "sha256:d433e1221f802dac393bc8652fabcc63aa46896cd920bb888ae0e2002fe6b756",
|
||||
"cuda10.0-cudnn7-centos7": "sha256:a453b7147a60928a8345689eae48916a746b3578b5e831bfa151f0529d469c88",
|
||||
"cuda10.0-cudnn7-centos6": "sha256:671757b382068522e2a94acacb35281c435e80964217173af103ee1b594fd4df",
|
||||
"cuda10.0-cudnn7-centos6": "sha256:a1909ba09c703340ee0074ce63dd94fe8fea48035a25264677907a609e2375e0",
|
||||
"cuda10.0-cudnn7-ubuntu16.04-manylinux2010": "sha256:eedcedfe63a778068bf725f9ffa425646725faac9ba96a57abfad307e832dcf9",
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user