前置条件:安装好相应版本cuda。
1、在conda中创建一个新的环境。
conda create -n conda-caffe python=3.5 numpy=1.13
2、安装必须的库。
sudo apt-get update
sudo apt-get upgrade
sudo apt-get install -y build-essential cmake git pkg-config
sudo apt-get install -y libprotobuf-dev libleveldb-dev libsnappy-dev protobuf-compiler libhdf5-serial-dev
sudo apt-get install -y libatlas-base-dev
sudo apt-get install -y --no-install-recommends libboost-all-dev
sudo apt-get install -y libgflags-dev libgoogle-glog-dev liblmdb-dev
3、clone或者直接下载源码。
4、进入caffe目录,创建build文件夹,然后使用cmake和make编译。
mkdir build && cd build
cmake -D python_version=3 ..
make all
make install
5、安装其他python数据处理包。
conda install cython scikit-image ipython h5py nose pandas protobuf pyyaml jupyter
6、安装完成,在每次使用conda中的caffe时,首先运行以下命令
① 激活conda环境:conda activate conda-caffe
② 临时导入PYTHONPATH:
export PYTHONPATH=$PYTHONPATH:/home/ubuntu/caffe/build/install/python
7、目前存在的问题:
只能使用python进行开发工作,在ipython中导入caffe时会报错,所以关于所有需要用到ipython的包暂时无法正常使用,其他暂未发现。
参考链接: