[Docker] update Dockerfile, add launch script

Dockerfile

@@ -2,8 +2,17 @@ FROM pytorch/pytorch:1.1.0-cuda10.0-cudnn7.5-devel
 MAINTAINER Yaman Umuroglu <yamanu@xilinx.com>
 ARG PYTHON_VERSION=3.6
 
-WORKDIR /workdir
-RUN git clone --branch feature/onnx_exec https://github.com/Xilinx/FINN.git finn
+WORKDIR /workspace
+RUN git clone https://github.com/maltanar/brevitas_cnv_lfc.git
+RUN git clone https://github.com/Xilinx/brevitas
+RUN cd brevitas; pip install .
 
-WORKDIR /workdir/finn
+COPY requirements.txt .
 RUN pip install -r requirements.txt
+RUN rm requirements.txt
+
+
+# Note that we expect the cloned finn directory on the host to be
+# mounted on /workspace/finn -- see run-docker.sh for an example
+# of how to do this.
+ENV PYTHONPATH "${PYTHONPATH}:/workspace/finn"

run-docker.sh

+#!/bin/sh
+
+# Absolute path to this script, e.g. /home/user/bin/foo.sh
+SCRIPT=$(readlink -f "$0")
+# Absolute path this script is in, thus /home/user/bin
+SCRIPTPATH=$(dirname "$SCRIPT")
+echo "Mounting $SCRIPTPATH into /workspace/finn"
+# Build the FINN Docker image
+docker build --tag=finn .
+# Launch container with current directory mounted
+docker run --rm --name finn_dev -it -v $SCRIPTPATH:/workspace/finn finn bash