Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
1 changed file
with
71 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,71 @@ | ||
FROM ubuntu:16.04 | ||
|
||
# install cuda | ||
RUN NVIDIA_GPGKEY_SUM=d1be581509378368edeec8c1eb2958702feedf3bc3d17011adbf24efacce4ab5 && \ | ||
NVIDIA_GPGKEY_FPR=ae09fe4bbd223a84b2ccfce3f60f4b3d7fa2af80 && \ | ||
apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub && \ | ||
apt-key adv --export --no-emit-version -a $NVIDIA_GPGKEY_FPR | tail -n +5 > cudasign.pub && \ | ||
echo "$NVIDIA_GPGKEY_SUM cudasign.pub" | sha256sum -c --strict - && rm cudasign.pub && \ | ||
echo "deb http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/cuda.list | ||
|
||
ENV CUDA_VERSION 8.0.61 | ||
LABEL com.nvidia.cuda.version="${CUDA_VERSION}" | ||
ENV NVIDIA_CUDA_VERSION $CUDA_VERSION | ||
|
||
ENV CUDA_PKG_VERSION 8-0=$CUDA_VERSION-1 | ||
RUN apt-get update && apt-get install -y --no-install-recommends \ | ||
cuda-nvrtc-$CUDA_PKG_VERSION \ | ||
cuda-nvgraph-$CUDA_PKG_VERSION \ | ||
cuda-cusolver-$CUDA_PKG_VERSION \ | ||
cuda-cublas-8-0=8.0.61.2-1 \ | ||
cuda-cufft-$CUDA_PKG_VERSION \ | ||
cuda-curand-$CUDA_PKG_VERSION \ | ||
cuda-cusparse-$CUDA_PKG_VERSION \ | ||
cuda-npp-$CUDA_PKG_VERSION \ | ||
cuda-cudart-$CUDA_PKG_VERSION && \ | ||
ln -s cuda-8.0 /usr/local/cuda && \ | ||
rm -rf /var/lib/apt/lists/* | ||
|
||
RUN echo "/usr/local/cuda/lib64" >> /etc/ld.so.conf.d/cuda.conf && \ | ||
ldconfig | ||
|
||
# nvidia-docker 1.0 | ||
LABEL com.nvidia.volumes.needed="nvidia_driver" | ||
|
||
RUN echo "/usr/local/nvidia/lib" >> /etc/ld.so.conf.d/nvidia.conf && \ | ||
echo "/usr/local/nvidia/lib64" >> /etc/ld.so.conf.d/nvidia.conf | ||
|
||
ENV PATH /usr/local/nvidia/bin:/usr/local/cuda/bin:${PATH} | ||
ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 | ||
|
||
# nvidia-container-runtime | ||
ENV NVIDIA_VISIBLE_DEVICES all | ||
ENV NVIDIA_DRIVER_CAPABILITIES compute,utility | ||
|
||
|
||
# install cudnn | ||
RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list | ||
|
||
ENV CUDNN_VERSION 6.0.21 | ||
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}" | ||
|
||
RUN apt-get update | ||
RUN apt-get install -y --no-install-recommends --allow-unauthenticated libcudnn6=$CUDNN_VERSION-1+cuda8.0 | ||
RUN rm -rf /var/lib/apt/lists/* | ||
|
||
|
||
# install python and essentials | ||
RUN apt-get update && apt-get -y upgrade | ||
RUN apt-get install -y python python-pip python-dev python-tk | ||
RUN apt-get install -y wget | ||
|
||
|
||
# setup working directory | ||
RUN mkdir -p /var/app | ||
WORKDIR /var/app | ||
COPY . /var/app | ||
|
||
|
||
# install dependencies | ||
RUN pip install --upgrade pip | ||
RUN pip install -r requirements.txt |