From f1858cfaedab74e12239a104284e0f5c74ae6551 Mon Sep 17 00:00:00 2001 From: Ishaan Datta Date: Wed, 11 Sep 2024 10:13:58 -0700 Subject: [PATCH] initial conversion containers --- Containers/PyTorch.Dockerfile | 2 +- Containers/TensorFlow.Dockerfile | 8 ++++++++ 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/Containers/PyTorch.Dockerfile b/Containers/PyTorch.Dockerfile index 656880c..a70b2ff 100644 --- a/Containers/PyTorch.Dockerfile +++ b/Containers/PyTorch.Dockerfile @@ -3,6 +3,6 @@ FROM nvcr.io/nvidia/pytorch:24.08-py3 COPY ../conversion_tools /conversion_tools # ORT -RUN pip3 install -U --no-cache-dir --verbose jetson-stats numpy onnx pycuda ultralytics +RUN pip3 install -U --no-cache-dir --verbose jetson-stats numpy onnx pycuda ultralytics argparse ENTRYPOINT [ "/bin/bash" ] \ No newline at end of file diff --git a/Containers/TensorFlow.Dockerfile b/Containers/TensorFlow.Dockerfile index e69de29..a2ca13f 100644 --- a/Containers/TensorFlow.Dockerfile +++ b/Containers/TensorFlow.Dockerfile @@ -0,0 +1,8 @@ +FROM nvcr.io/nvidia/tensorflow:24.08-tf2-py3 + +COPY ../conversion_tools /conversion_tools + +# ORT +RUN pip3 install -U --no-cache-dir --verbose jetson-stats numpy onnx pycuda argparse + +ENTRYPOINT [ "/bin/bash" ] \ No newline at end of file