写在前面
NVIDIA caffe 是NVIDIA维护的caffe的一个分支,有很多便于部署在边缘端的特性,详细信息去NVIDIA的github看吧:https://github.com/NVIDIA/caffe
jetson_nano 安装这个步骤非常繁琐,这里记录一下。以下两段代码来自李丹丹老师所著书《基于NVIDIA jetson平台的人工智能实例开发入门》,你可以看这本书,这里写出是以备自己的书不在旁边。
安装Protobuf
安装依赖项:
sudo apt install -y autoconf automake libtool curl make g++ git python-dev python-setuptools unzip
下载安装protobuf:
git clone https://github.com/google/protobuf -b '3.2.x'cd protobuf
sudo ./autogen.sh
sudo ./configure
sudo make -j6
sudo make install
sudo ldconfig
安装python运行库:
cd python
sudo python setup.py install --cpp_implementation
安装NVIDIA-caffe
安装依赖项:
sudo apt install -y build-essential cmake git gfortran libopenblas-dev libboost-all-dev libgflags-dev libgoogle-glog-dev libhdf5-serial-dev libleveldb-dev liblmdb-dev libopencv-dev libsnappy-dev python-all-dev python-dev python-h5py python-matplotlib python-numpy python-opencv python-pil python-pip python-pydot python-scipy python-skimage python-sklearn libturbojpeg0-dev libfreetype6-dev
下载caffe:
进行到这里,有个错误需要纠正 就是要下载最新的NVIDIA-caffe 而非书中提供的版本 不然编译会出错
git clone https://github.com/NVIDIA/caffe
安装运行库以及caffe:
sudo -H pip install -r ~/caffe/python/requirements.txtcd caffe
mkdir build
cd build
cnake ..
sudo make -j6
sudo make install
设置环境变量:
nano ~/.bashrc
export PYTHONPATH=/home/yu/caffe/python
source ~/.bashrc
有可能会遇到的问题:
https://blog.csdn.net/qq_35550465/article/details/81194602
sudo make pycaffe
https://www.cnblogs.com/chay/p/10273927.html
安装DIGITS
安装依赖项:
sudo apt install --no-install-recommends git graphviz python-dev python-flask python-flaskext.wtf python-gevent python-h5py python-numpy python-pil python-scipy python-tk
下载安装DIGITS及其python运行库
git clone https://github.com/NVIDIA/Digitssudo -H pip install -r Digits/requirements.txtsudo -H pip install -e Digits/
配置环境变量:
sudo nano ~/.bashrcexport CAFFE_ROOT=刚刚安装caffe的路径source ~/.bashrc
有一处坑是flask-wtf版本问题
sudo -H pip install flask-wtf==0.14.3 -i https://pypi.doubanio.com/simple
flask-wtf 要安装最新版本 2020年2月26日 为0.14.3