【解决】Failed to get convolution algorithm. This is probably because cuDNN failed to initialize.
报错描述:
配置yolov4-deepsort官方代码环境,依据官方测试运行命令:
python save_model.py --model yolov4
出现报错:
dependencies:
- python==3.7
- cudnn==7.5.1
- cudatoolkit==10.1.243
- tensorflow-gpu==2.3.0rc0
- opencv-python==4.1.1.26
解决方法:
一、如果运行的代码中已经包含了:
①使用tensorflow:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as session:
② 使用keras:
config = tf.ConfigProto()
config.gpu_options.allow_growth = True
keras.backend.tensorflow_backend.set_session(tf.Session(config=config))
二、还是不行的话,就去检查和尝试实际运行的代码环境中,cuddn+ cuda + python的版本是不是匹配的。
这里,调试使用了: cuddn-v10.0-v7.5.1+ cuda10.0 + python3.7
后,可以成功加速运行代码。
运行效果:
Reference
https://blog.csdn.net/qq_41868689/article/details/98503069
.https://blog.csdn.net/weixin_45617478/article/details/107884222