modelzoo training mode error

yessine · ‎2024-09-20

im trying to train a model based on pascal 2012 dataset (training mode). after the system laod the datasets. when it starts training 1/50 epoque i encounter several errors about cudnn(please check image) im using cudnn v9.4 . i found that tensorflow 2.8.3 (modelzoo work with) is compatible only with cudnn v8.1 is it true. and how to install v8.1 because when i download teh v8.1 from nvidia website . i dont find the setup file to execute.

Julian E. · ‎2024-09-25

Hello @yessine ,

Something probably went wrong when downloading the requirements for ST Model Zoo.

You may want to reinstall everything. Please read the Before you start section: https://github.com/STMicroelectronics/stm32ai-modelzoo?tab=readme-ov-file#before-you-start

Let me know if you need further help.

Julian

In order to give better visibility on the answered topics, please click on 'Accept as Solution' on the reply which solved your issue or answered your question.