Skip to content
This repository has been archived by the owner on Dec 7, 2023. It is now read-only.

Commit

Permalink
Added VPI test
Browse files Browse the repository at this point in the history
  • Loading branch information
rbonghi committed Feb 6, 2023
1 parent 3932c7d commit 6d29868
Show file tree
Hide file tree
Showing 2 changed files with 68 additions and 0 deletions.
44 changes: 44 additions & 0 deletions vpi/Dockerfile.vpi
Original file line number Diff line number Diff line change
@@ -0,0 +1,44 @@
FROM nvidia/cuda:11.4.1-devel-ubuntu20.04

ARG L4T=35.1

# Disable terminal interaction for apt
ENV DEBIAN_FRONTEND=noninteractive
ENV SHELL /bin/bash
SHELL ["/bin/bash", "-c"]

########### Sources and main Jetson repo ###################

RUN apt-get update \
&& apt-get install -y --no-install-recommends software-properties-common \
&& apt-key adv --fetch-key https://repo.download.nvidia.com/jetson/jetson-ota-public.asc \
&& if [ "$(uname -m)" = "x86_64" ]; then \
# Adding sources for discrete NVIDIA GPU
add-apt-repository "deb http://repo.download.nvidia.com/jetson/x86_64/focal r${L4T} main"; \
else \
# Adding sources for NVIDIA Jetson
add-apt-repository "deb https://repo.download.nvidia.com/jetson/common r${L4T} main"; \
fi

########### VPI ############################################

RUN apt-get update \
&& apt-get install -y \
libegl1 libnvvpi2 vpi2-dev vpi2-samples \
g++ cmake libopencv-dev \
&& rm -rf /var/lib/apt/lists/* \
&& apt-get clean

ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/opt/nvidia/vpi2/lib64"
ENV LD_LIBRARY_PATH="${LD_LIBRARY_PATH}:/usr/local/cuda-11.4/targets/x86_64-linux/lib/stubs:/usr/local/cuda-11.4/targets/sbsa-linux/lib/stubs"

############################################################

RUN cp -r /opt/nvidia/vpi2/samples /root/NVIDIA_VPI-2.1-samples

WORKDIR /root/NVIDIA_VPI-2.1-samples/01-convolve_2d

RUN cmake .

ARG SKIP_MAKE
RUN [ "${SKIP_MAKE}" != "" ] || make
24 changes: 24 additions & 0 deletions vpi/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,24 @@
# Testing VPI

This test docker is made to check the VPI capabilities for a NVIDIA Jetpack


## Jetpack 5.0.2

Build

```bash
docker build --build-arg L4T=35.1 -t vpi-test -f Dockerfile.vpi .
```

Should crash!

```bash
docker build --build-arg L4T=35.1 --build-arg SKIP_MAKE=yes -t vpi-test-run -f Dockerfile.vpi .
```

## Jetpack 5.1

```bash
docker build --build-arg L4T=35.2 -t vpi-test -f Dockerfile.vpi .
```

0 comments on commit 6d29868

Please sign in to comment.