diff --git a/Dockerfile.tmpl b/Dockerfile.tmpl
index 10faf2ce..5423c240 100644
--- a/Dockerfile.tmpl
+++ b/Dockerfile.tmpl
@@ -512,6 +512,7 @@ RUN pip install flashtext && \
     # ipympl adds interactive widget support for matplotlib
     pip install ipympl==0.7.0 && \
     pip install pandarallel && \
+    pip install onnx && \
     /tmp/clean-layer.sh
 
 # Download base easyocr models.
diff --git a/tests/data/mnist-8.onnx b/tests/data/mnist-8.onnx
new file mode 100644
index 00000000..fc1a3f73
Binary files /dev/null and b/tests/data/mnist-8.onnx differ
diff --git a/tests/test_onnx.py b/tests/test_onnx.py
new file mode 100644
index 00000000..c09cffc9
--- /dev/null
+++ b/tests/test_onnx.py
@@ -0,0 +1,8 @@
+import unittest
+
+import onnx
+
+class TestOnnx(unittest.TestCase):
+    def test_load(self):
+        model = onnx.load("/input/tests/data/mnist-8.onnx")
+        self.assertIn("CNTKGraph", onnx.helper.printable_graph(model.graph))