Update super_resolution_with_onnxruntime.py (#2647)

add install for onnx Fixes #2646 Description Install ONNX to avoid dependency issues --------- Co-authored-by: Nikita Shulga <2453524+malfet@users.noreply.github.com>
pytorch · Nov 5, 2023 · 77aec05 · 77aec05
1 parent e7563f6
commit 77aec05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/advanced_source/super_resolution_with_onnxruntime.py b/advanced_source/super_resolution_with_onnxruntime.py
@@ -26,7 +26,7 @@
 .. code-block:: bash
 
    %%bash
-   pip install onnxruntime
+   pip install onnx onnxruntime
 
 ONNX Runtime recommends using the latest stable runtime for PyTorch.