[Docker] add docker image for cuda10.0

Author: jiazhihao

docker/Dockerfile

@@ -0,0 +1,45 @@
+# Copyright 2019 Stanford
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+FROM nvidia/cuda:10.0-devel-ubuntu16.04
+
+RUN apt-get update && apt-get install -y --no-install-recommends wget sudo binutils git && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN wget -c http://developer.download.nvidia.com/compute/redist/cudnn/v7.6.0/cudnn-10.0-linux-x64-v7.6.0.64.tgz && \
+    tar -xzf cudnn-10.0-linux-x64-v7.6.0.64.tgz -C /usr/local && \
+    rm cudnn-10.0-linux-x64-v7.6.0.64.tgz && \
+    ldconfig
+
+RUN wget -c https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
+    mv Miniconda3-latest-Linux-x86_64.sh ~/Miniconda3-latest-Linux-x86_64.sh && \
+    chmod +x ~/Miniconda3-latest-Linux-x86_64.sh && \
+    ~/Miniconda3-latest-Linux-x86_64.sh -b -p /opt/conda && \
+    rm ~/Miniconda3-latest-Linux-x86_64.sh && \
+    /opt/conda/bin/conda upgrade --all && \
+    /opt/conda/bin/conda install conda-build conda-verify && \
+    /opt/conda/bin/conda clean -ya
+
+RUN /opt/conda/bin/conda install cmake make
+RUN /opt/conda/bin/conda install -c conda-forge protobuf=3.9 numpy onnx
+RUN /opt/conda/bin/conda install -c anaconda cython
+
+ENV PATH /opt/conda/bin:$PATH
+
+COPY install_taso.sh /install/install_taso.sh
+RUN bash /install/install_taso.sh
+
+WORKDIR /usr/TASO
+#RUN chmod -R a+w /usr/TASO

docker/install_taso.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+# Copyright 2019 Stanford
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+set -e
+set -u
+set -o pipefail
+
+cd /usr
+git clone --recursive https://github.com/jiazhihao/TASO.git
+cd /usr/TASO
+mkdir -p build
+cd build
+cmake ..
+sudo make install -j10
+cd /usr/TASO/python
+python setup.py install

docker/run_docker.sh

@@ -0,0 +1,29 @@
+#!/usr/bin/env bash
+# Copyright 2019 Stanford
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+IMAGE_NAME=("$1")
+COMMAND="bash"
+
+WORKSPACE="/usr/TASO"
+
+echo "WORKSPACE: ${WORKSPACE}"
+echo "IMAGE NAME: ${IMAGE_NAME}"
+echo "DOCKER BINARY: nvidia-docker"
+
+nvidia-docker run --rm --pid=host \
+    -it --net=host \
+    ${IMAGE_NAME} \
+    ${COMMAND[@]}

include/taso/ops.h

@@ -1,4 +1,4 @@
-/* Copyright 2018 Stanford
+/* Copyright 2019 Stanford
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.