kws Dockerfile

Dockerfile

FROM nvidia/cuda:7.5-runtime-ubuntu14.04
LABEL maintainer "NVIDIA CORPORATION <[email protected]>"

RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1404/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list

ENV CUDNN_VERSION 4.0.8
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN apt-get update && apt-get install -y --no-install-recommends \
            libcudnn4=$CUDNN_VERSION && \
    rm -rf /var/lib/apt/lists/*
RUN apt-get update
RUN apt-get install -y python python-dev python-virtualenv libsamplerate-dev libfftw3-3 wget nano cmake
# Set the working directory to /kws
WORKDIR /kws
# Copy the current directory contents into the container at /kws
ADD . /kws
RUN virtualenv kws
RUN source kws/bin/activate
RUN pip install pip --upgrade
RUN pip install -r requirements.txt
RUN pip install tensorflow
RUN make
RUN export PYTHONPATH=$PYTHONPATH:`pwd`
RUN mkdir /models
RUN sed -i -e 's/home\/ubuntu\/models\/kws\/test/models\/test/' ./configs/kws.json