tensorflow和caffe 的docker安装过程及相关linux操作总结

一、caffe  docker的安装

• docker pull caffe镜像
• 安装依赖包
• 使用docker安装省去安装CUDA和cudnn的安装。
• 安装opencv3（使用源码安装）

下载opencv源码到 /opt 目录下，解压并进入opencv-3.1.0目录;

mkdir build # 创建编译的文件目录

cd build cmake -D CMAKE_BUILD_TYPE=Release -D CMAKE_INSTALL_PREFIX=/usr/local ..

make -j8 #编译

??? sudo make install #安装

///////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////

        pkg-config --modversion opencv

        pkg-config --libs --cflags opencv #查看gcc或g++编译的CFLAGS，即头文件路径和动态库路径

        注意的是静态库使用ar命令对 .o 文件操作创建的，文件以 .a 结束命名; 使用时在静态库名前加 --lib，如gcc -o hello main.c -L. -lmyhello

        同理动态库使用 gcc -shared -fPCI -o libmyhello.so hello.o 命令创建，文件以 .so 命名;

1、使用时记得把 libmyhello.so 复制到目录 /usr/lib 中再跑 gcc -o hello main.c -L. -lmyhello; 因为一般Linux系统把/lib和/usr/lib两个目录作为默认的库搜索路径

2、或者添加linux的搜索路径，在 /etc/ld.so.conf 文件中添加库的搜索路径；当静态库和动态库同名时， gcc命令将优先使用动态库。

       如果 pkg-config --modversion查找不到版本，则手动添加opencv源码的opencv4.pc文件到/usr/lib/pkconfig/;

       opencv4.pc的目录在 ~/Desktop/opencv-4.0.0-alpha/build/unix-install，可以使用 find ./ -name opencv4.pc 来在当前目录往下查找

       或者使用 export PKG_CONFIG_PATH=~/Desktop/opencv-4.0.0-alpha/build/unix-install:$PKG_CONFIG_PATH //opencv4.pc在源码的路径

然后使用$echo$PKG_CONFIG_PATH 来查看 PKG_CONFIG_PATH的路径设置是什么。

• 安装caffe

git clone https://github.com/BVLC/caffe.git

sudo cp Makefile.config.example Makefile.config

sudo gedit Makefile.config #开启选项和设置路径（cudnn，opencv 版本，python 接口层，python 路径）


然后修改 caffe 目录下的 Makefile 文件（NVCCFLAGS，LIBRARIES）

make all -j8? #编译


sudo make runtest -j8? #跑自带的测试

###编译pycaffe

cd caffe

sudo make pycaffe -j8

?

• 下载模型VGG16预训练模型：

　　sh  data/scripts/fetch_faster_rcnn_models.sh

 

• caffe安装编译成功后，跑fast-cnn代码时（Python tools/demo.py），会出现cudnn6不兼容该项目的错误，此时有两种方法：

1. cd caffe-fast-rcnn
   git remote add caffe https://github.com/BVLC/caffe.git
   git fetch caffe
   git merge caffe/master
   最后注释掉 self_.attr(“phase”) = static_cast(this->phase_); from include/caffe/layers/python_layer.hpp after merging.
   这种方法有时候会成功，有时候则不会
2. 手动修改文件
   1. 用最新caffe源码的以下文件替换掉faster rcnn 的对应文件
   include/caffe/layers/cudnn_relu_layer.hpp, src/caffe/layers/cudnn_relu_layer.cpp, src/caffe/layers/cudnn_relu_layer.cu
   include/caffe/layers/cudnn_sigmoid_layer.hpp, src/caffe/layers/cudnn_sigmoid_layer.cpp, src/caffe/layers/cudnn_sigmoid_layer.cu
   include/caffe/layers/cudnn_tanh_layer.hpp, src/caffe/layers/cudnn_tanh_layer.cpp, src/caffe/layers/cudnn_tanh_layer.cu
   2 用caffe源码中的这个文件替换掉faster rcnn 对应文件
   include/caffe/util/cudnn.hpp

 至此可以使用训练好的模型VGG16来检测识别物体了。

 

 

docker命令

docker rename 原容器名  新容器名

docker tag 41b730702607 opmaster/basic_server:devel //镜像id  ，改新镜像名字和标签

//////////////////////////

#使docker映射到本机host的GUI口来显示，使用nvidia的GPU来创建容器，并提供交互（-it）的bash，设置数据卷来挂载共享数据，并命名容器名

sudo nvidia-docker run -it
> -v /tmp/.X11-unix:/tmp/.X11-unix
> -e DISPLAY=unix$DISPLAY
> -e GDK_SCALE
> -e GDK_DPI_SCALE
> -v /home/csj/Desktop/caffe_docker_mnt/:/home/caffe_docker_mnt
> --name shijie_caffe_container1
> 1febdeef0276
> /bin/bash

 

jupyter notebook --allow-root   #使用jupyter notebook

////////////////////////

sudo nvidia-docker start   2806cb9d9633

sudo nvidia-docker exec -it  2806cb9d9633 /bin/bash
sudo nvidia-docker commit 2806cb9d9633 shijie_image:v1 //保存改动的容器到镜像

docker commit  -m="First Image"  -a="786389209"  7a15f99695c0   786389209/unbantu:17.10.0 #

* -m 提交的描述信息

* -a 指定镜像作者

sudo nvidia-docker save shijie_image:v1   /home/shijie_docker/shijie_image.tar //保存镜像为 .tar 文件到host本地目录

sudo nvidia-docker load   -i   file-path

#注意load和import命令不同，前者导入镜像到本地镜像仓库，后者只是导入一个快照，退出会回滚原始版本。

rm          #删除一个或多个容器

rmi         #删除一个或多个镜像

 

#登录docker并上传镜像

docker  login

docker  push  786389209/unbantu-test:lastest

 

 

cmake 的介绍

　　CMake是一个比make更高级的编译配置工具，它可以根据不同平台、不同的编译器，生成相应的Makefile或者vcproj项目。通过编写CMakeLists.txt，可以控制生成的Makefile，从而控制编译过程。CMake自动生成的Makefile不仅可以通过make命令构建项目生成目标文件，还支持安装（make install）、测试安装的程序是否能正确执行（make test，或者ctest）、生成当前平台的安装包（make package）、生成源码包（make package_source）、产生Dashboard显示数据并上传等高级功能，只要在CMakeLists.txt中简单配置，就可以完成很多复杂的功能，包括写测试用例。如果有嵌套目录，子目录下可以有自己的CMakeLists.txt。总之很强大！

比如Kdevelop中opencv的文件简单的CMakeLists.txt：

 

cmake_minimum_required(VERSION 2.8)
project(opencv)  #工程名字
find_package(OpenCV REQUIRED)
SET(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -std=c++11")  #opencv4必须添加对C++11的支持

add_executable(opencv main.cpp)
target_link_libraries(opencv ${OpenCV_LIBS})

 

【说明】：本文章由站长整理发布，文章内容不代表本站观点，如文中有侵权行为，请与本站客服联系(QQ:254677821)！