Merge pull request #4230 from caisq/fix-docker-build-1
Fix nightly docker builds
This commit is contained in:
commit
90581c5844
@ -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 && \
|
||||||
|
@ -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
|
||||||
|
@ -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 && \
|
||||||
|
Loading…
Reference in New Issue
Block a user