adding docker support (airctic#895)

* adding docker support

* simplifying Dockerfile

Author: FraPochetti

docker/Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.8
+
+RUN pip install torch==1.8.0+cu101 torchvision==0.9.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html -U
+RUN pip install git+git://github.com/airctic/icevision.git#egg=icevision[all] -U
+RUN pip install git+git://github.com/airctic/icedata.git -U
+RUN pip install yolov5-icevision -U 
+RUN pip install mmcv-full==1.3.7 -f https://download.openmmlab.com/mmcv/dist/cu101/torch1.8.0/index.html -U
+RUN pip install mmdet==2.13.0 -U
+RUN pip install ipywidgets

docker/README.md

@@ -0,0 +1,2 @@
+## How-to
+For instructions on how to use the docker-related files in this folder check out [this post](https://francescopochetti.com/developing-inside-a-docker-container-in-visual-studio-code/).

docker/devcontainer.json

@@ -0,0 +1,11 @@
+{
+	"name": "Icevision",
+	"dockerComposeFile": ["docker-compose.yaml"],
+	"service": "icevision",
+	"runServices": ["icevision"],
+	"workspaceFolder": "/root",
+	"extensions": [
+		"ms-python.python",
+		"ms-azuretools.vscode-docker"
+	],
+}

docker/docker-compose.gpu.yaml

@@ -0,0 +1,11 @@
+version: "3"
+
+services:
+  icevision:
+    deploy:
+      resources:
+        reservations:
+          devices:
+          - driver: nvidia
+            count: 1
+            capabilities: [gpu]

docker/docker-compose.yaml

@@ -0,0 +1,11 @@
+version: "3"
+
+services:
+  icevision:
+    build:
+      dockerfile: Dockerfile
+      context: .
+    image: ice
+    tty: true
+    volumes:
+      - ../:/root/