首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >在Colab中找不到DNN库ssd_mobile_net_v2

在Colab中找不到DNN库ssd_mobile_net_v2
EN

Stack Overflow用户
提问于 2022-02-05 13:50:54
回答 2查看 5.3K关注 0票数 5

我在用Tensorflow训练一个手检测模型我有个问题。

这是训练模型的代码:

代码语言:javascript
运行
复制
!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

我有一个错误:

代码语言:javascript
运行
复制
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]
EN

回答 2

Stack Overflow用户

发布于 2022-05-27 11:15:31

我能够通过运行以下命令来解决这个问题。

代码语言:javascript
运行
复制
!apt install --allow-change-held-packages libcudnn8=8.1.0.77-1+cuda11.2
票数 4
EN

Stack Overflow用户

发布于 2022-02-07 13:12:20

据我所见,这与较新版本的TensorFlow (2.8.0)有关。我能够通过遵循- https://github.com/tensorflow/models/issues/10479#issuecomment-1029626502来解决这个问题。

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70998639

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档