Added docker container for support DeepStack in Nvidia Jetson Orin AGX 64GB Developer Kit - Jetpack 6.1 #37
Workflow file for this run
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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 | |