diff --git a/Dockerfile.arm64 b/Dockerfile.arm64 new file mode 100644 index 0000000000000000000000000000000000000000..147fdeb2acacbfffef81c651d0bb7f3920ffda94 --- /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 a472ce90c32a32442add42e08db658c4a6aa00c4..a5406c0adfeee62eec680e413e9189f5d0665ada 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