Skip to content

Commit

Permalink
Fix gstreamer import position (#725)
Browse files Browse the repository at this point in the history
  • Loading branch information
denisvmedyantsev committed Apr 8, 2024
1 parent 3b8209d commit 63682a0
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 3 deletions.
2 changes: 1 addition & 1 deletion docker/Dockerfile.deepstream
Original file line number Diff line number Diff line change
Expand Up @@ -291,7 +291,7 @@ RUN wget -nv -O ${PYCUDA_WHL} ${PACKAGES_URL}/${PYCUDA_WHL} && \

# onnx, onnxruntime (use extra-index to get the version with CUDA 12.2 support)
ARG ORT_VERSION=1.17.1
RUN python -m pip install onnx onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/
RUN python -m pip install --no-cache-dir onnx onnxruntime-gpu --extra-index-url https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/

# cython, polars, scikit-learn, jupyter
RUN python -m pip install --no-cache-dir \
Expand Down
6 changes: 4 additions & 2 deletions savant/entrypoint/main.py
Original file line number Diff line number Diff line change
@@ -1,18 +1,20 @@
"""Module entrypoints."""
"""Module entrypoints.
isort:skip_file
"""
import os
import signal
from pathlib import Path
from threading import Thread
from typing import IO, Any, Union

from savant.gstreamer import Gst # should be first
from savant.config import ModuleConfig
from savant.config.schema import ElementGroup, ModelElement
from savant.deepstream.encoding import check_encoder_is_available
from savant.deepstream.nvinfer.build_engine import build_engine
from savant.deepstream.nvinfer.model import NvInferModel
from savant.deepstream.pipeline import NvDsPipeline
from savant.deepstream.runner import NvDsPipelineRunner
from savant.gstreamer import Gst
from savant.healthcheck.server import HealthCheckHttpServer
from savant.healthcheck.status import ModuleStatus, set_module_status
from savant.utils.check_display import check_display_env
Expand Down

0 comments on commit 63682a0

Please sign in to comment.