30 lines
1.2 KiB
Plaintext
30 lines
1.2 KiB
Plaintext
|
FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04
|
||
|
|
|
||
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||
|
|
python3-pip python3-dev python3-setuptools \
|
||
|
|
&& \
|
||
|
|
apt-get clean && \
|
||
|
|
rm -rf /var/lib/apt/lists/* \
|
||
|
|
&& pip3 install tensorflow-gpu==1.4.1
|
||
|
|
|
||
|
|
RUN pip3 list && pip3 install numpy boto3 six awscli flask==0.11 Jinja2==2.9 gevent gunicorn keras==2.1.3 pillow h5py \
|
||
|
|
&& pip3 list
|
||
|
|
|
||
|
|
# Configure the build for our CUDA configuration.
|
||
|
|
ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs:$LD_LIBRARY_PATH
|
||
|
|
ENV CI_BUILD_PYTHON=python \
|
||
|
|
LD_LIBRARY_PATH=/usr/local/cuda/extras/CUPTI/lib64:$LD_LIBRARY_PATH \
|
||
|
|
TF_NEED_CUDA=1 \
|
||
|
|
TF_CUDA_VERSION=8.0 \
|
||
|
|
TF_CUDNN_VERSION=6 \
|
||
|
|
TF_CUDA_COMPUTE_CAPABILITIES=3.7,6.1
|
||
|
|
|
||
|
|
# Fix paths so that CUDNN can be found
|
||
|
|
# See https://github.com/tensorflow/tensorflow/issues/8264
|
||
|
|
RUN ls -lah /usr/local/cuda/lib64/*
|
||
|
|
RUN mkdir /usr/lib/x86_64-linux-gnu/include/ && \
|
||
|
|
ln -s /usr/lib/x86_64-linux-gnu/include/cudnn.h /usr/lib/x86_64-linux-gnu/include/cudnn.h && \
|
||
|
|
ln -s /usr/include/cudnn.h /usr/local/cuda/include/cudnn.h && \
|
||
|
|
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so /usr/local/cuda/lib64/libcudnn.so && \
|
||
|
|
ln -s /usr/lib/x86_64-linux-gnu/libcudnn.so.6 /usr/local/cuda/lib64/libcudnn.so.6
|