diff options
author | Martin Polden <mpolden@mpolden.no> | 2023-02-09 12:11:21 +0100 |
---|---|---|
committer | Martin Polden <mpolden@mpolden.no> | 2023-02-09 12:12:20 +0100 |
commit | e02adcebeb90661046893a2c386e330cdc925327 (patch) | |
tree | 317427d7f6ed1af49205017670ac87fc2c2089b3 /model-integration | |
parent | 5c5a79813ea6d6d57326f82145816200b8a41622 (diff) |
Fix GPU detection
Diffstat (limited to 'model-integration')
-rw-r--r-- | model-integration/src/main/java/ai/vespa/modelintegration/evaluator/OnnxEvaluator.java | 2 |
1 files changed, 1 insertions, 1 deletions
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); } |