torch
onnx
opencv-python
numpy
faiss-gpu
pycuda
tensorrt
dask-cuda-cu12 --extra-index-url=https://pypi.nvidia.com
pycuda
