Dockerfile

Dockerfile

@@ -1,12 +1,10 @@
-FROM  tensorflow/tensorflow:2.5.1
+FROM python:3.7.16
 
-RUN apt-get install make g++ gcc
-RUN pip3 install gunicorn
-RUN pip3 install setuptools==46.1.3
+RUN pip install gunicorn==20.1.0
+RUN pip install setuptools==46.1.3
 
 RUN mkdir -p /app
 WORKDIR /app
-
 COPY requirements.txt /app
 RUN pip install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
 

Dockerfile_arm

@@ -0,0 +1,16 @@
+FROM nvcr.io/nvidia/l4t-tensorflow:r32.6.1-tf2.5-py3
+RUN apt-get install make g++ gcc
+RUN pip3 install gunicorn
+RUN pip3 install setuptools==46.1.3
+
+RUN mkdir -p /app
+WORKDIR /app
+
+COPY requirements.txt /app
+RUN pip3 install -r requirements.txt -i https://mirrors.aliyun.com/pypi/simple/
+
+COPY ./app /app
+EXPOSE 5000
+CMD ["gunicorn", "--bind", ":5000", "server:app"]
+
+

docker-compose.yml

@@ -1,6 +1,6 @@
 version: "3.6"
 services:
-  hpds-road-detection-model:
+  hpds-python-model:
     container_name: hpds-road-detection-model
     image: hpds-road-detection:1.0.0 
     networks:

requirements.txt

@@ -1,4 +1,5 @@
 Flask==1.1.2
-numpy==1.18.4
+numpy==1.19.5
 Pillow==7.1.2
 six==1.15.0
+tensorflow==2.5.1