使用tensorflow2.8.0加载YOLOv3,默认使用cpu,速度较慢。
随后安装cuda11.6,cudnn8.3,出现如下错误。
Successfully load weights!
2022-07-05 14:33:05.710450: I tensorflow/stream_executor/cuda/cuda_dnn.cc:368] Loaded cuDNN version 8302
Could not load library cudnn_cnn_infer64_8.dll. Error code 126
Please make sure cudnn_cnn_infer64_8.dll is in your library path!
解决方案:
安装并配置zlib
下载链接:http://www.winimage.com/zLibDll/zlib123dllx64.zip
随后将libcuda.so文件找到并加入PATH,
重启电脑即可。
saved_model_full模型运行成功
利用test样例检验成功