Downgrade default ONNX Runtime to CPU

Dockerfile

@@ -18,6 +18,8 @@ WORKDIR /code
 COPY ./ /code
 
 RUN pip3 install --no-cache-dir -r requirements.txt \
- && pip3 install -e .
+ && pip3 install -e . \
+ && pip3 uninstall -y onnxruntime \
+ && pip3 install onnxruntime-gpu
 
 CMD python3 app.py

README.rst

@@ -24,6 +24,13 @@ or, from source:
    cd scoutbot
    (.venv) $ pip install -e .
 
+To then add GPU acceleration, you need to replace `onnxruntime` with `onnxruntime-gpu`:
+
+.. code-block:: console
+
+   (.venv) $ pip uninstall -y onnxruntime
+   (.venv) $ pip install onnxruntime-gpu
+
 How to Run
 ----------
 

requirements.txt

@@ -3,7 +3,7 @@ cryptography
 gradio
 imgaug
 numpy
-onnxruntime-gpu
+onnxruntime
 opencv-python-headless
 Pillow
 pooch

scoutbot/__init__.py

@@ -56,7 +56,7 @@ log = utils.init_logging()
 
 from scoutbot import agg, loc, tile, wic  # NOQA
 
-VERSION = '0.1.8'
+VERSION = '0.1.9'
 version = VERSION
 __version__ = VERSION
 

setup.cfg

@@ -23,7 +23,7 @@ install_requires =
     gradio
     imgaug
     numpy
-    onnxruntime-gpu
+    onnxruntime
     opencv-python-headless
     Pillow
     pooch