重训练ssd_mobilenet时TensorFlow导致系统崩溃,请求技术支持
SSD-Mobilenet目标检测训练环境与终端记录
我整理了这套用于SSD-Mobilenet目标检测训练的环境配置细节,以及对应的Putty终端会话片段,方便你参考:
系统与训练环境配置
- 操作系统:Ubuntu 16.04
- GPU硬件:GTX 1050 TI
- NVIDIA驱动版本:390.30
- CUDA版本:9.1
- cuDNN版本:7.0.5
- TensorFlow-GPU版本:1.7rc1 及 1.5
- 训练文件说明:配置文件与
train.py均为TensorFlow官方分发的原始文件 - 预训练模型:
ssd_mobilenet_v1_coco_2017_11_17
Putty终端会话片段
(od) gennis@AI:~/models/research$ ./train_raccoon.sh WARNING:tensorflow:From /home/dennis/.virtualenvs/od/local/lib/python2.7/site-packages/tensorfl...
内容的提问来源于stack exchange,提问作者Gennis
相关产品推荐
相关产品推荐

