-
Notifications
You must be signed in to change notification settings - Fork 0
/
Dockerfile
21 lines (17 loc) 路 1000 Bytes
/
Dockerfile
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu18.04
LABEL maintainer="Bryan Flood <bryanfloodcontact@gmail.com>"
LABEL description="馃惓 Docker environment for Swift GPU Accelerated Machine Learning"
LABEL url="https://github.com/KnowledgePending/Swift-TensorFlow-with-CUDA-Docker"
RUN apt-get -qq update && \
apt-get -qq upgrade && \
apt-get -qq install clang \
libpython-dev \
libblocksruntime-dev \
wget \
software-properties-common && \
add-apt-repository ppa:ubuntu-toolchain-r/test && \
apt-get -qq upgrade libstdc++6
RUN wget https://storage.googleapis.com/swift-tensorflow-artifacts/releases/v0.4/rc2/swift-tensorflow-RELEASE-0.4-cuda10.0-cudnn7-ubuntu18.04.tar.gz && \
tar xvzf swift-tensorflow-RELEASE-0.4-cuda10.0-cudnn7-ubuntu18.04.tar.gz && \
rm swift-tensorflow-RELEASE-0.4-cuda10.0-cudnn7-ubuntu18.04.tar.gz && \
export PATH=$(path)/usr/bin:"${PATH}"