Add support for a docker container for Jetson boards
This commit is contained in:
13
Dockerfile.arm64
Normal file
13
Dockerfile.arm64
Normal file
@@ -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 . .
|
||||
@@ -4,6 +4,7 @@ services:
|
||||
pipeline:
|
||||
build:
|
||||
context: .
|
||||
dockerfile: ${DOCKERFILE:-Dockerfile}
|
||||
command:
|
||||
- python3
|
||||
- s2s_pipeline.py
|
||||
|
||||
Reference in New Issue
Block a user