ONNXRuntimeError

ERROR - root - 执行训练过程失败: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : 初始化期间发生异常: /onnxruntime_src/onnxruntime/core/providers/cuda/cuda_call.cc:121 std::conditional_t<THRW, void, onnxruntime::common::Status> onnxruntime::CudaCall(ERRTYPE, const char*, const char*, ERRTYPE, const char*, const char*, int) [with ERRTYPE = cublasStatus_t; bool THRW = true; std::conditional_t<THRW, void, onnxruntime::common::Status> = void] /onnxruntime_src/onnxruntime/core/providers/cuda/cuda_call.cc:114 std::conditional_t<THRW, void, onnxruntime::common::Status> onnxruntime::CudaCall(ERRTYPE, const char*, const char*, ERRTYPE, const char*, const char*, int) [with ERRTYPE = cublasStatus_t; bool THRW = true; std::conditional_t<THRW, void, onnxruntime::common::Status> = void] CUBLAS 失败 3: CUBLAS_STATUS_ALLOC_FAILED; GPU=0; 主机名=egg; 文件=/onnxruntime_src/onnxruntime/core/providers/cuda/cuda_execution_provider.cc; 行=164; 表达式=cublasCreate(&cublas_handle_);