mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-14 20:48:00 +00:00
parent
9d80235607
commit
9318b3a47f
1 changed files with 1 additions and 1 deletions
|
|
@ -107,7 +107,7 @@ class Validator : public OutputCollector<TCharString> {
|
|||
void CreateSession() {
|
||||
Ort::SessionOptions session_options;
|
||||
#ifdef USE_CUDA
|
||||
Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CUDA(session_option, 0));
|
||||
Ort::ThrowOnError(OrtSessionOptionsAppendExecutionProvider_CUDA(session_options, 0));
|
||||
#endif
|
||||
session_ = Ort::Session(env_, model_path_.c_str(), session_options);
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue