Skip to content

Added docker container for support DeepStack in Nvidia Jetson Orin AGX 64GB Developer Kit - Jetpack 6.1 #37

Added docker container for support DeepStack in Nvidia Jetson Orin AGX 64GB Developer Kit - Jetpack 6.1

Added docker container for support DeepStack in Nvidia Jetson Orin AGX 64GB Developer Kit - Jetpack 6.1 #37

Workflow file for this run

name: DevTest
on:
push:
branches:
- dev
- master
pull_request:
branches:
- dev
- master
jobs:
docker-gpu-build:
runs-on: [self-hosted, Linux, X64, gpu]
steps:
- name: Checkout repo
uses: actions/checkout@master
- name: Fetch GIT LFS
run: git lfs checkout && git lfs pull
- name: Install Pytest
run: pip3 install pytest
- name: Install Wget
run: pip3 install wget
- name: Install PIL
run: pip3 install pillow
- name: Download Binary Dependencies
run: python3 download_dependencies.py
- name: Clean DeepStack Container
run: docker container rm -f deepstack
continue-on-error: true
- name: Build DeepStack GPU Version
run: docker build -t deepquestai/deepstack:gpu . -f Dockerfile.gpu --build-arg tag=gpu-1647248158
- name: Test Scene
run: |
docker run --gpus all -d --name deepstack -e VISION-SCENE=True -p 80:5000 deepquestai/deepstack:gpu
cd tests
export TEST_DATA_DIR=$PWD"/test_data"
export TEST_DEEPSTACK_URL="http://localhost:80"
cd scene
python3 -m pytest
docker container rm -f deepstack
- name: Test Detection
run: |
docker run --gpus all -d --name deepstack -e VISION-DETECTION=True -p 80:5000 deepquestai/deepstack:gpu
cd tests
export TEST_DATA_DIR=$PWD"/test_data"
export TEST_DEEPSTACK_URL="http://localhost:80"
cd detection
python3 -m pytest
docker container rm -f deepstack
- name: Test Face
run: |
docker run --gpus all -d --name deepstack -e VISION-FACE=True -p 80:5000 deepquestai/deepstack:gpu
cd tests
export TEST_DATA_DIR=$PWD"/test_data"
export TEST_DEEPSTACK_URL="http://localhost:80"
cd face
python3 -m pytest
docker container rm -f deepstack
- name: Test Custom Detection
run: |
cd tests
modeldir=$(pwd)
docker run --gpus all -d --name deepstack -v $modeldir/"test_data":/modelstore/detection -p 80:5000 deepquestai/deepstack:gpu
export TEST_DATA_DIR=$PWD"/test_data"
export TEST_DEEPSTACK_URL="http://localhost:80"
cd custom
python3 -m pytest
docker container rm -f deepstack
- name: Test Enhance
run: |
docker run --gpus all -d --name deepstack -e VISION-ENHANCE=True -p 80:5000 deepquestai/deepstack:gpu
cd tests
export TEST_DATA_DIR=$PWD"/test_data"
export TEST_DEEPSTACK_URL="http://localhost:80"
cd enhance
python3 -m pytest
docker container rm -f deepstack
- name: Clean DeepStack GPU Version
run: docker image rm -f deepquestai/deepstack:gpu