From d8afa799ac72c3411ab27846af4e0f50137a187f Mon Sep 17 00:00:00 2001
From: Andres Marafioti <andimarafioti@gmail.com>
Date: Tue, 10 Sep 2024 14:06:05 +0200
Subject: [PATCH] Add support for a docker container for Jetson boards
---
Dockerfile.arm64 | 13 +++++++++++++
docker-compose.yml | 1 +
2 files changed, 14 insertions(+)
create mode 100644 Dockerfile.arm64
diff --git a/Dockerfile.arm64 b/Dockerfile.arm64
new file mode 100644
index 0000000..147fdeb
--- /dev/null
+++ b/Dockerfile.arm64
@@ -0,0 +1,13 @@
+FROM nvcr.io/nvidia/l4t-pytorch:r35.2.1-pth2.0-py3
+
+ENV PYTHONUNBUFFERED 1
+
+WORKDIR /usr/src/app
+
+# Install packages
+RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
+
+COPY requirements.txt ./
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY . .
\ No newline at end of file
diff --git a/docker-compose.yml b/docker-compose.yml
index a472ce9..a5406c0 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -4,6 +4,7 @@ services:
pipeline:
build:
context: .
+ dockerfile: ${DOCKERFILE:-Dockerfile}
command:
- python3
- s2s_pipeline.py
--
GitLab