Add a tutorial notebook for ONNX inference

Dockerfile

@@ -170,4 +170,10 @@ COPY ./rastervision_pytorch_learner/ /opt/src/rastervision_pytorch_learner/
 RUN pip install sagemaker_pytorch_training==2.8.1
 ENV SAGEMAKER_TRAINING_MODULE=sagemaker_pytorch_container.training:main
 
+# Install a onnxruntime-gpu version compatible with CUDA 12. Specifying
+# --extra-index-url in requirements.txt seems to cause problems witht the
+# RTD build.
+RUN pip uninstall onnxruntime-gpu -y && \
+    pip install --upgrade onnxruntime-gpu==1.17 --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
+
 CMD ["bash"]

docs/conf.py

@@ -140,6 +140,7 @@ def setup(app: 'Sphinx') -> None:
     # more natural img/tensorboard.png path does not work.
     'usage/tutorials/train': '_images/tensorboard.png',
     'usage/tutorials/lightning_workflow': '_images/lightning-logo.png',
+    'usage/tutorials/onnx_inference': '_images/ONNX_Runtime_logo_dark.png',
 }
 nbsphinx_prolog = r"""
 {% set docpath = env.doc2path(env.docname, base=False) %}

docs/usage/tutorials/index.rst

@@ -24,3 +24,4 @@ The following tutorials highlight many of the things you can do with Raster Visi
     prechipped_datasets
     stac_plus_osm
     temporal
+    onnx_inference