diff --git a/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt b/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt index d7a2cd4dd..89f00109a 100644 --- a/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt +++ b/Tools/PyTorch/TimeSeriesPredictionPlatform/triton/requirements.txt @@ -16,7 +16,7 @@ natsort>=7.0.0 networkx==2.5 numpy onnx>=1.8.0,<1.9.0 -onnxruntime-gpu==1.8.1 +onnxruntime-gpu==1.18.1 pycuda>=2019.1.2 PyYAML>=5.2 tabulate>=0.8.7