我在用Tensorflow训练一个手检测模型我有个问题。
这是训练模型的代码:
!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
我有一个错误:
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]
发布于 2022-05-27 11:15:31
我能够通过运行以下命令来解决这个问题。
!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2
发布于 2022-02-07 13:12:20
据我所见,这与较新版本的TensorFlow (2.8.0)有关。我能够通过遵循- https://github.com/tensorflow/models/issues/10479#issuecomment-1029626502来解决这个问题。
https://stackoverflow.com/questions/70998639
复制相似问题