search

HaxyMoly / Pangu-Weather-ReadyToGo

盘古天气大模型全流程演示（ 输入数据准备、预测及结果可视化）Unofficial demonstration of Huawei's Pangu Weather Model. Implementing the entire process of data preparation for input, forecasting conversion of forecasted results, and visualization.
159 stars 32 forks source link

bug #17

Open bravelyi opened 7 months ago

bravelyi commented 7 months ago

Traceback (most recent call last): File "C:\Users\Administrator\Desktop\code\qx\Pangu-Weather-ReadyToGo-main\inference.py", line 104, in <module> ort_session = ort.InferenceSession(model_used, sess_options=options, providers=[('CUDAExecutionProvider', cuda_provider_options)]) File "D:\Anaconda3\envs\pangu\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 360, in init self._create_inference_session(providers, provider_options, disabled_optimizers) File "D:\Anaconda3\envs\pangu\lib\site-packages\onnxruntime\capi\onnxruntime_inference_collection.py", line 408, in _create_inference_session sess.initialize_session(providers, provider_options, disabled_optimizers) onnxruntime.capi.onnxruntime_pybind11_state.RuntimeException: [ONNXRuntimeError] : 6 : RUNTIME_EXCEPTION : Exception during initialization: D:\a_work\1\s \onnxruntime\core\providers\cuda\cuda_call.cc:124 onnxruntime::CudaCall D:\a_work\1\s\onnxruntime\core\providers\cuda\cuda_call.cc:117 onnxruntime::CudaCall CUDNN failure 1: CUDNN_STATUS_NOT_INITIALIZED ; GPU=0 ; hostname=WIN-PS05E4J14Q1 ; expr=cudnnCreate(&cudnnhandle);

bravelyi commented 7 months ago

你们有人在运行inference.py时遇到这个问题吗

smarttommyau commented 7 months ago

你是用了gpu版的吗？可能你安装onnxruntime 有问题