diff --git a/server/setup.py b/server/setup.py index 1f7d81d2..b59f8b6a 100644 --- a/server/setup.py +++ b/server/setup.py @@ -54,7 +54,11 @@ 'onnx', 'onnxmltools', ] - + (['onnxruntime-gpu'] if sys.platform != 'darwin' else ['onnxruntime']), + + ( + ['onnxruntime-gpu<=1.13.1'] + if sys.platform != 'darwin' + else ['onnxruntime<=1.13.1'] + ), 'tensorrt': [ 'nvidia-tensorrt==8.4.1.5', ],