This repository was archived by the owner on Jun 10, 2024. It is now read-only.
File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 3535 # already installed on runner: git build-essential python3-dev python3-pip cmake (new version required, use pip if toold)
3636 - name : Install dependencies
3737 if : ${{ matrix.os != 'windows-2022' }}
38- run : |
38+ run : |
3939 sudo apt-get update
4040 sudo apt-get install \
4141 libavfilter-dev \
@@ -49,11 +49,11 @@ jobs:
4949
5050 - name : Install dependencies (Windows)
5151 if : ${{ matrix.os == 'windows-2022' }}
52- run : |
52+ run : |
5353 curl -L https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-win64-lgpl-shared.zip --output ffmpeg.zip
5454 unzip ffmpeg.zip
5555 mv ffmpeg*-shared ffmpeg
56-
56+
5757 # CUDA toolkit: the following is sufficient on Ubuntu cuda-libraries-XX-X cuda-compiler-XX-X libnpp-dev-XX-X libnvidia-decode-XXX
5858 - uses : Jimver/cuda-toolkit@v0.2.8
5959 id : cuda-toolkit
@@ -74,15 +74,16 @@ jobs:
7474
7575 - name : Install CUVID
7676 if : ${{ matrix.os != 'windows-2022' }}
77- run : |
77+ run : |
7878 # normally you would install them via `apt install cuda`
7979 sudo apt-get install \
8080 libnpp-dev-11-7 \
8181 libnvidia-decode-520 \
8282 -y
8383
8484 - name : Build and Install Package
85- run : |
85+ run : |
86+ CUDA_VERSION=11.7.0 \
8687 SKBUILD_CONFIGURE_OPTIONS="-DTC_FFMPEG_ROOT=D:/a/VideoProcessingFramework/VideoProcessingFramework/ffmpeg/" python -mpip install . --verbose
8788
8889 - name : Check Import
Original file line number Diff line number Diff line change @@ -17,22 +17,22 @@ jobs:
1717 # The CMake configure and build commands are platform agnostic and should work equally well on Windows or Mac.
1818 # You can convert this to a matrix build if you need cross-platform coverage.
1919 # See: https://docs.github.com/en/free-pro-team@latest/actions/learn-github-actions/managing-complex-workflows#using-a-build-matrix
20- runs-on : self-hosted
20+ runs-on : self-hosted
2121
2222 steps :
2323 - uses : actions/checkout@v3
2424 - name : Build docker containers
25- run : |
25+ run : |
2626 DOCKER_BUILDKIT=1 docker build -f docker/Dockerfile --tag vpf-gpu --build-arg PIP_INSTALL_EXTRAS=torch .
2727 DOCKER_BUILDKIT=1 docker build -f docker/Dockerfile.tensorrt --build-arg PIP_INSTALL_EXTRAS=samples --tag vpf-gpu-trt .
2828
2929 - name : Run tests
30- run : |
30+ run : |
3131 docker run --gpus all --rm --user $UID -v $GITHUB_WORKSPACE:/cwd --entrypoint "python3" vpf-gpu -m unittest discover /cwd/tests
3232
3333 - name : Run samples
34- run : |
34+ run : |
3535 docker run --gpus all --rm --user $UID -v $GITHUB_WORKSPACE:/cwd \
3636 -e TORCH_HOME=/cwd/cache \
37+ -e HOME=/cwd \
3738 --entrypoint "make" vpf-gpu-trt run_samples_without_docker
38-
You can’t perform that action at this time.
0 commit comments