-
Notifications
You must be signed in to change notification settings - Fork 13
/
Copy pathDockerfile
30 lines (25 loc) · 1.03 KB
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
# Ubuntu 16.04, CUDA 9.0
FROM nvidia/cuda:9.0-runtime-ubuntu16.04
LABEL maintainer "MICROSOFT CORPORATION" \
com.microsoft.cntk.version="2.4"
ENV CNTK_VERSION="2.4"
RUN apt-get update && apt-get install -y --no-install-recommends \
# General
ca-certificates \
wget \
sudo \
build-essential \
&& \
# Clean-up
apt-get -y autoremove \
&& \
rm -rf /var/lib/apt/lists/*
# Get CNTK Binary Distribution
RUN CNTK_VERSION_DASHED=$(echo $CNTK_VERSION | tr . -) && \
CNTK_SHA256="8eabc27f91d3ac7a10c6a089f6eab9d019808ac4510e52aee982f61d560df897" && \
wget -q https://cntk.ai/BinaryDrop/CNTK-${CNTK_VERSION_DASHED}-Linux-64bit-GPU.tar.gz && \
echo "$CNTK_SHA256 CNTK-${CNTK_VERSION_DASHED}-Linux-64bit-GPU.tar.gz" | sha256sum --check --strict - && \
tar -xzf CNTK-${CNTK_VERSION_DASHED}-Linux-64bit-GPU.tar.gz && \
rm -f CNTK-${CNTK_VERSION_DASHED}-Linux-64bit-GPU.tar.gz && \
/bin/bash /cntk/Scripts/install/linux/install-cntk.sh --py-version 35 --docker
WORKDIR /root