Merge pull request #52 from idavis/feature/jp-4.4.1

JetPack 4.4.1 and Nano 2GB Support

Makefile

@@ -32,9 +32,11 @@ export DOCKERFILE_PREFIX ?= default
 
 all: jetpack-deps driver-packs jetpacks
 
-driver-packs: driver-pack-32.4.3 driver-pack-32.3.1 driver-pack-32.2.3 driver-pack-32.2.1 driver-pack-32.2.0 driver-pack-32.1
+driver-packs: driver-pack-32.4.4 driver-pack-32.4.3 driver-pack-32.3.1 driver-pack-32.2.3 driver-pack-32.2.1 driver-pack-32.2.0 driver-pack-32.1
 
 
+driver-pack-32.4.4: l4t-32.4.4-tx1 l4t-32.4.4-jax l4t-32.4.4-jax-8gb l4t-32.4.4-tx2 l4t-32.4.4-nano-dev l4t-32.4.4-nano l4t-32.4.4-nano-2gb-dev l4t-32.4.4-tx2i l4t-32.4.4-tx2-4gb l4t-32.4.4-nx-dev l4t-32.4.4-nx
+
 driver-pack-32.4.3: l4t-32.4.3-tx1 l4t-32.4.3-jax l4t-32.4.3-jax-8gb l4t-32.4.3-tx2 l4t-32.4.3-nano-dev l4t-32.4.3-nano l4t-32.4.3-tx2i l4t-32.4.3-tx2-4gb l4t-32.4.3-nx-dev l4t-32.4.3-nx
 
 driver-pack-32.3.1: l4t-32.3.1-tx1 l4t-32.3.1-jax l4t-32.3.1-jax-8gb l4t-32.3.1-tx2 l4t-32.3.1-nano-dev l4t-32.3.1-nano l4t-32.3.1-tx2i l4t-32.3.1-tx2-4gb
@@ -74,6 +76,9 @@ from-deps-folder-%:
 # JetPack
 
 
+32.4.4-%:
+	make -C $(CURDIR)/docker/jetpack $@
+
 32.4.3-%:
 	make -C $(CURDIR)/docker/jetpack $@
 

docker/jetpack/4.4.1/jax-8gb/all/Dockerfile

@@ -0,0 +1,129 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-devel-cudnn
+
+# NVIDIA VisionWorks Toolkit
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks libvisionworks-dev libvisionworks-samples && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# NVIDIA VisionWorks Plus (SFM)
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks-sfm libvisionworks-sfm-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# NVIDIA VisionWorks Object Tracker
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks-tracking libvisionworks-tracking-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+
+# NVIDIA TensorRT
+
+
+LABEL com.nvidia.libinfer.version="7.1.3"
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libnvinfer7 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libnvinfer-bin libnvinfer-dev libnvinfer-doc libnvinfer-samples && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libnvinfer-plugin-dev libnvinfer-plugin7 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+
+LABEL com.nvidia.tensorrt.version="7.1.3.0"
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends tensorrt && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libnvparsers7 libnvparsers-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libnvonnxparsers7 libnvonnxparsers-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends python-libnvinfer python-libnvinfer-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends python3-libnvinfer python3-libnvinfer-dev && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Graph Surgeon
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends graphsurgeon-tf && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# UFF Converter
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends uff-converter-tf && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Multimedia API
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends nvidia-l4t-jetson-multimedia-api && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+
+# Install dependencies for OpenCV
+
+# Open CV 4.1.1
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libopencv libopencv-dev libopencv-python libopencv-samples opencv-licenses && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install dependencies for TensorFlow
+#RUN apt-get update && apt-get install -y \
+#    hdf5-tools \
+#    libhdf5-dev \
+#    libhdf5-serial-dev \
+#    libjpeg8-dev \
+#    zip \
+#    zlib1g-dev \
+#    && \
+#    apt-get clean && \
+#    rm -rf /var/lib/apt/lists/*
+
+#RUN python3 -m pip install -U numpy grpcio absl-py py-cpuinfo psutil portpicker grpcio six mock requests gast h5py astor termcolor
+
+# Install TensorFlow
+# Index URLS
+# Browse https://developer.download.nvidia.com/compute/redist/jp/ for more.
+# 4.4: https://developer.download.nvidia.com/compute/redist/jp/v44/tensorflow/
+# 4.3 GPU: https://developer.download.nvidia.com/compute/redist/jp/v43/tensorflow-gpu/
+# 4.3 CPU: https://developer.download.nvidia.com/compute/redist/jp/v43/tensorflow/
+# 4.2: https://developer.download.nvidia.com/compute/redist/jp/v42/tensorflow-gpu/
+# Commands:
+# RUN python3 -m pip install --pre --extra-index-url ${TF_INDEX_URL} tensorflow-gpu
+# RUN python3 -m pip install --extra-index-url ${TF_INDEX_URL} tensorflow-gpu==${TF_VERSION}+nv${NV_VERSION}
+
+# Available versions:
+# All available versions can be found by browsing https://developer.download.nvidia.com/compute/redist/jp/

docker/jetpack/4.4.1/jax-8gb/base/Dockerfile

@@ -0,0 +1,32 @@
+
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb
+
+LABEL com.nvidia.jetpack.version="4.4.1"
+
+ENV CUDA_VERSION "10.2.89"
+
+LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends --allow-downgrades cuda-cudart-10-2 cuda-nvtx-10-2 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+RUN ln -s /usr/local/cuda-10.2 /usr/local/cuda && \
+    ln -s /usr/local/cuda-10.2/targets/aarch64-linux/include /usr/local/cuda/include && \
+    ln -s /usr/local/cuda-10.2/targets/aarch64-linux/lib /usr/local/cuda/lib64
+
+RUN echo "/usr/lib/aarch64-linux-gnu/tegra" >> /etc/ld.so.conf.d/nvidia-tegra.conf && \
+    echo "/usr/lib/aarch64-linux-gnu/tegra-egl" >> /etc/ld.so.conf.d/nvidia-tegra.conf && \
+    echo "/usr/local/cuda-10.2/targets/aarch64-linux/lib" >> /etc/ld.so.conf.d/nvidia.conf && \
+    ldconfig
+
+RUN export LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH} && \
+    export LIBRARY_PATH=/usr/local/cuda/lib64/stubs && \
+    grep -q "export PATH=.*/usr/local/cuda/bin" ~/.bashrc || echo "export PATH=/usr/local/cuda/bin:$PATH">>~/.bashrc && \
+    grep -q "export LD_LIBRARY_PATH=/usr/local/cuda/lib64" ~/.bashrc || echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH" >> ~/.bashrc && \
+    grep -q "export PATH=.*/usr/local/cuda/bin" /etc/skel/.bashrc || echo "export PATH=/usr/local/cuda/bin:$PATH">>/etc/skel/.bashrc && \
+    grep -q "export LD_LIBRARY_PATH=/usr/local/cuda/lib64" /etc/skel/.bashrc || echo "export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH" >> /etc/skel/.bashrc
+
+ENV NVIDIA_VISIBLE_DEVICES all
+ENV NVIDIA_DRIVER_CAPABILITIES all

docker/jetpack/4.4.1/jax-8gb/deepstream/Dockerfile

@@ -0,0 +1,91 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-base
+
+# NVIDIA VisionWorks Toolkit
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# NVIDIA VisionWorks Plus (SFM)
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks-sfm && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# NVIDIA VisionWorks Object Tracker
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libvisionworks-tracking && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# NVIDIA TensorRT
+
+
+LABEL com.nvidia.tensorrt.version="7.1.3.0"
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends tensorrt && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Graph Surgeon
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends graphsurgeon-tf && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# UFF Converter
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends uff-converter-tf && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Open CV 4.1.1
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libopencv && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# DeepStream Dependencies
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    libssl1.0.0 \
+    libgstreamer1.0-0 \
+    gstreamer1.0-tools \
+    gstreamer1.0-plugins-good \
+    gstreamer1.0-plugins-bad \
+    gstreamer1.0-plugins-ugly \
+    gstreamer1.0-libav \
+    libgstrtspserver-1.0-0 \
+    libjansson4 \
+    libjson-glib-1.0-0 \
+    && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# Additional DeepStream dependencies usually installed by the CUDA Toolkit
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    libgstreamer1.0-dev \
+    libgstreamer-plugins-base1.0-dev \
+    && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+# DeepStream
+
+
+RUN apt-get update && apt-get install -y --no-install-recommends deepstream-5.0=5.0.1-1 && \
+
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+RUN ldconfig
+WORKDIR /opt/nvidia/deepstream/deepstream-4.0/samples

docker/jetpack/4.4.1/jax-8gb/devel/Dockerfile

@@ -0,0 +1,9 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-runtime
+
+# CUDA Toolkit for L4T
+
+RUN apt-get update && \
+    apt-get install -y --allow-downgrades --no-install-recommends cuda-toolkit-10-2 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*

docker/jetpack/4.4.1/jax-8gb/devel/cudnn/Dockerfile

@@ -0,0 +1,13 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-devel
+
+# NVIDIA CUDA Deep Neural Network library (cuDNN)
+
+ENV CUDNN_VERSION 8.0
+
+LABEL com.nvidia.cudnn.version="8.0"
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libcudnn8 libcudnn8-dev libcudnn8-doc && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*

docker/jetpack/4.4.1/jax-8gb/runtime/Dockerfile

@@ -0,0 +1,7 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-base
+
+# CUDA libraries for L4T
+RUN apt-get update && apt-get install -y --no-install-recommends cuda-libraries-10.2 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*

docker/jetpack/4.4.1/jax-8gb/runtime/cudnn/Dockerfile

@@ -0,0 +1,13 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-runtime
+
+# NVIDIA CUDA Deep Neural Network library (cuDNN)
+ENV CUDNN_VERSION 8.0
+
+LABEL com.nvidia.cudnn.version="8.0"
+
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends libcudnn8 && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*

docker/jetpack/4.4.1/jax-8gb/samples/Dockerfile

@@ -0,0 +1,35 @@
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-devel as builder
+
+WORKDIR /usr/local/cuda-10.2/samples
+
+RUN make -j4
+
+ARG IMAGE_NAME
+FROM ${IMAGE_NAME}:32.4.4-jax-8gb-jetpack-4.4.1-runtime as base
+
+# Prereqs
+
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    build-essential \
+    freeglut3 \
+    libegl1 \
+    libx11-dev \
+    libgles2-mesa \
+    libgl1-mesa-glx \
+    libglu1-mesa \
+    libgomp1 \
+    libxi-dev \
+    libxmu-dev \
+    openmpi-bin \
+    && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+FROM base
+RUN mkdir samples
+COPY --from=builder /usr/local/cuda-10.2/samples/ /samples
+
+
+WORKDIR /samples/bin/aarch64/linux/release/