-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile.nvidia
20 lines (16 loc) · 1.1 KB
/
Dockerfile.nvidia
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# ARG repository
FROM nvidia/cuda:8.0-devel-ubuntu16.04
LABEL maintainer "NVIDIA CORPORATION <[email protected]>"
RUN echo "deb https://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 && \
apt-get install -y wget bzip2 vim && \
apt-get install -y --no-install-recommends libcudnn6=$CUDNN_VERSION-1+cuda8.0 libcudnn6-dev=$CUDNN_VERSION-1+cuda8.0 && \
apt-get install -y build-essential && \
apt-get install -y cmake git libgtk2.0-dev pkg-config libavcodec-dev libavformat-dev libswscale-dev && \
apt-get install -y python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev libjasper-dev libdc1394-22-dev && \
apt-get install -y pkg-config zip g++ zlib1g-dev unzip python && \
rm -rf /var/lib/apt/lists/* && \
wget https://repo.continuum.io/archive/Anaconda3-2018.12-Linux-x86_64.sh && \
wget https://github.com/bazelbuild/bazel/releases/download/0.18.0/bazel-0.18.0-installer-linux-x86_64.sh