From e02adcebeb90661046893a2c386e330cdc925327 Mon Sep 17 00:00:00 2001 From: Martin Polden Date: Thu, 9 Feb 2023 12:11:21 +0100 Subject: Fix GPU detection --- .../main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'model-integration/src') diff --git a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java index 563ef911f8f..5cc7991d197 100644 --- a/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java +++ b/model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java @@ -101,7 +101,7 @@ public class OnnxEvaluator { return createSession(modelPath, environment, options, false); } if (isCudaError(e)) { - throw new IllegalArgumentException("GPU device is requested, but CUDA initialization failed", e); + throw new IllegalArgumentException("GPU device is required, but CUDA initialization failed", e); } throw new RuntimeException("ONNX Runtime exception", e); } -- cgit v1.2.3