Skip to main content
Associate III
September 20, 2024
Question

modelzoo training mode error

  • September 20, 2024
  • 1 reply
  • 1037 views

Screenshot 2024-09-20 193733.pngim 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. 

1 reply

Julian E.
Technical Moderator
September 25, 2024

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.