search

pipeless-ai / pipeless

An open-source computer vision framework to build and deploy apps in minutes
https://pipeless.ai
Apache License 2.0
688 stars 31 forks source link

fix(container): Use base image with CUDA 11.7 for TensorRT #125

Closed miguelaeh closed 5 months ago

miguelaeh commented 5 months ago

Description of the change

The TensorRT image was using a base image with CUDA 12, producing an error because the ONNX runtime was unable to find the CUDA 11 libraries. This PR fixes the problem by using a base image that ships CUDA 11.7, which is also consistent with the CUDA container image.