Testing bayer decoding on gpu.

2025-01-14 17:56:36 +01:00
parent 2f52521b9a
commit b804ce149d
4 changed files with 253 additions and 1 deletions
@@ -34,5 +34,8 @@ RUN pip3 install --upgrade --no-cache-dir onnxconverter_common
 # Fix an error when profiling
 RUN pip3 install --upgrade --no-cache-dir "onnxruntime-gpu<1.17"

+RUN pip3 install --upgrade --no-cache-dir tensorflow
+RUN pip3 install --upgrade --no-cache-dir tf2onnx
+
 WORKDIR /mmdeploy/
 CMD ["/bin/bash"]