首页 / LINUX / SSD的训练 Linux下
SSD的训练 Linux下
内容导读
互联网集市收集整理的这篇技术教程文章主要介绍了SSD的训练 Linux下,小编现在分享给大家,供广大互联网技能从业者学习和参考。文章包含1242字,纯文字阅读大概需要2分钟。
内容图文
1.下载caffe-ssd
cd ssd # ssd是一个空的文件
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd #检查分支
2.编译前准备:Makefile.config
cp Makefile.config.example Makefile.config
不谈版本的编译都不是好编译!!
Python3.5+cudnn6.0.21+cuda8.0.61+opencv3.30
打开Makefile.config:
USE_CUDNN := 1
OPENCV_VERSION := 3
CUDA_DIR := /usr/local/cuda
CUDA_ARCH := -gencode arch=compute_30,code=sm_30 -gencode arch=compute_35,code=sm_35 -gencode arch=compute_50,code=sm_50 -gencode arch=compute_52,code=sm_52 -gencode arch=compute_60,code=sm_60 -gencode arch=compute_61,code=sm_61 -gencode arch=compute_61,code=compute_61
BLAS := atlas
PYTHON_LIBRARIES := boost_python-py35 python3.5m
PYTHON_INCLUDE := /usr/include/python3.5m \ /usr/lib/python3.5/dist-packages/numpy/core/include \ /usr/local/lib/python3.5/dist-packages/numpy/core/include
PYTHON_LIB := /usr/lib
WITH_PYTHON_LAYER := 1
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
USE_NCCL := 1
BUILD_DIR := build
DISTRIBUTE_DIR := distribute
TEST_GPUID := 0 Q ?= @
2.编译:
make all -j8
make runtest -j8
make pycaffe -j8
内容总结
以上是互联网集市为您收集整理的SSD的训练 Linux下全部内容,希望文章能够帮你解决SSD的训练 Linux下所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错,欢迎将互联网集市网站推荐给程序员好友。
内容备注
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 gblab@vip.qq.com 举报,一经查实,本站将立刻删除。
内容手机端
扫描二维码推送至手机访问。
来源:【匿名】