DNN library is not found ssd_mobile_net_v2 in Colab

user_17530449 0
Viewed 4802

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]
keras tensorflow python
2 Answers

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
user_145063070
user_113137080
Related