Merge pull request #4230 from caisq/fix-docker-build-1

Fix nightly docker builds
This commit is contained in:
caisq 2016-09-06 12:57:04 -04:00 committed by GitHub
commit 90581c5844
3 changed files with 5 additions and 4 deletions

View File

@ -13,6 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
python \ python \
python-dev \ python-dev \
rsync \ rsync \
software-properties-common \
unzip \ unzip \
&& \ && \
apt-get clean && \ apt-get clean && \

View File

@ -85,8 +85,10 @@ RUN git clone -b r0.10 --recursive --recurse-submodules https://github.com/tenso
WORKDIR /tensorflow WORKDIR /tensorflow
# Configure the build for our CUDA configuration. # Configure the build for our CUDA configuration.
ENV CUDA_PATH /usr/local/cuda
ENV CUDA_TOOLKIT_PATH /usr/local/cuda ENV CUDA_TOOLKIT_PATH /usr/local/cuda
ENV CUDNN_INSTALL_PATH /usr/local/cuda ENV CUDNN_INSTALL_PATH /usr/lib/x86_64-linux-gnu
ENV LD_LIBRARY_PATH /usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64
ENV TF_NEED_CUDA 1 ENV TF_NEED_CUDA 1
RUN ./configure && \ RUN ./configure && \
@ -96,9 +98,6 @@ RUN ./configure && \
WORKDIR /root WORKDIR /root
# Set up CUDA variables
ENV CUDA_PATH /usr/local/cuda
# TensorBoard # TensorBoard
EXPOSE 6006 EXPOSE 6006
# IPython # IPython

View File

@ -13,6 +13,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
python \ python \
python-dev \ python-dev \
rsync \ rsync \
software-properties-common \
unzip \ unzip \
&& \ && \
apt-get clean && \ apt-get clean && \