-
Notifications
You must be signed in to change notification settings - Fork 656
Open
Labels
questionFurther information is requestedFurther information is requested
Description
Describe the question.
Hi! I have nvidia-dali pipeline where i try to decode video on GPU.
I have nvidia-dali-cuda120==1.51.2
My pipeline looks next way
import nvidia.dali.fn as fn
import nvidia.dali.types as types
from nvidia.dali.plugin.triton import autoserialize
import os
from nvidia.dali import pipeline_def
OUTPUT_DIR = os.path.dirname(os.path.abspath(__file__))
@pipeline_def(batch_size=1, num_threads=2, device_id=0)
def video_decode_pipeline():
video_data = fn.external_source(name="DALI_INPUT_0", device="cpu", dtype=types.UINT8)
frames = fn.experimental.decoders.video(video_data, device="mixed")
return frames
if __name__ == "__main__":
pipe = video_decode_pipeline(device_id=0)
output_fn = os.path.join(OUTPUT_DIR, "model.dali")
pipe.serialize(filename=output_fn)
With config.pbtxt
name: "keyframes_dali"
backend: "dali"
max_batch_size: 1
input [
{
name: "DALI_INPUT_0"
data_type: TYPE_UINT8
dims: [ -1 ]
}
]
output [
{
name: "DALI_OUTPUT_0"
data_type: TYPE_UINT8
dims: [ -1, -1, 3 ]
}
]
instance_group [
{
kind: KIND_GPU
count: 1
}
]
It successfully loaded to Triton Inference Server but when i try to infer model with bytes of video file i get next error
InferenceServerException: [StatusCode.UNKNOWN] Runtime error: Critical error in pipeline:
Error when executing Mixed operator experimental__decoders__Video encountered:
Error in thread 0: CUDA driver API error CUDA_ERROR_SHARED_OBJECT_SYMBOL_NOT_FOUND (302):
shared object symbol not found
Current pipeline object is no longer valid.
I tried to look for this error but had no success in it. Can you help me what should i check?
Check for duplicates
- I have searched the open bugs/issues and have found no duplicates for this bug report
Metadata
Metadata
Assignees
Labels
questionFurther information is requestedFurther information is requested