'DNN library is not found ssd_mobile_net_v2 in Colab

I'm training a hand detection model with Tensorflow and I have a problem.

This is the code to train the model:

!cd RealTimeObjectDetection/RealTimeObjectDetection && python Tensorflow/models/research/object_detection/model_main_tf2.py --model_dir=/content/RealTimeObjectDetection/RealTimeObjectDetection/Tensorflow/workspace/models/my_ssd_mobnet --pipeline_config_path=Tensorflow/workspace/models/my_ssd_mobnet/pipeline.config --num_train_steps=5000

I have this error:

Node: 'ssd_mobile_net_v2_fpn_keras_feature_extractor/model/Conv1/Conv2D'
DNN library is not found.
     [[{{node ssd_mobile_net_v2_fpn_keras_feature_extractor/model/Conv1/Conv2D}}]] [Op:__inference__dummy_computation_fn_15081]

Solution 1:^[1]

From what I can find, this has to do with the newer version of TensorFlow (2.8.0). I was able to resolve the issue by following this - https://github.com/tensorflow/models/issues/10479#issuecomment-1029626502

Solution 2:^[2]

I was able to resolve the issue by running the following command.

!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2

Sources

This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.

Source: Stack Overflow

Solution	Source
Solution 1	user7440787
Solution 2	puchal

'DNN library is not found ssd_mobile_net_v2 in Colab

Solution 1:[1]

Solution 2:[2]

Sources

Related Questions

Solution 1:^[1]

Solution 2:^[2]