原创博客,转载请说明出处!
Ubuntu 16.04+CUDA 9.1+cuDNN v7+OpenCV 3.4.0+Caffe的配置见我的另一篇博文,这边博文讲解配置好caffe之后如何使其在PyCharm上运行。配置过程分为以下六步:
1、安装依赖库
2、配置环境变量
3、编译 pycaffe
4、配置PyCharm所需环境
5、安装PyCharm
6、设置PyCharm解释器
以下为详细配置过程:
1、安装依赖库
为了防止遗漏,把下面的都装一下,已经装过的不会重复安装的,防止后面出现找不到模块的问题,在终端依次输入如下命令:
pip install -U scikit-image #若没有安装pip: sudo apt install python-pip
sudo apt-get install libprotobuf-dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5-serial-dev protobuf-compiler
sudo apt-get install --no-install-recommends libboost-all-dev
sudo apt-get install python-numpy python-scipy python-matplotlib python-sklearn python-skimage python-h5py python-protobuf python-leveldb python-networkx python-nose python-pandas python-gflags
sudo apt-get install protobuf-c-compiler protobuf-compiler
sudo apt-get install libgflags-dev libgoogle-glog-dev liblmdb-dev
2、配置环境变量
使用 gedit 命令打开配置文件:
sudo gedit ~/.bashrc
在文件最下面写入(导入caffe/python文件夹的路径,检查一下此路径是否和我的相同):
export PYTHONPATH=~/caffe/python:$PYTHONPATH
不做这一步 import caffe的时候会报错:找不到caffe模块。保存退出,并在终端写入下面语句,使环境变量生效:
source ~/.bashrc
3、编译 pycaffe
cd 进入caffe文件夹,运行:
sudo make pycaffe -j8
编译 pycaffe 成功后,验证一下是否可以在 python 中导入 caffe 包,首先进入 python 环境,直接在命令行输入:
python
然后导入 caffe :
>>> import caffe
若不报错则表示 caffe 的 python 接口已正确编译。假如报错的话执行一下 “4、配置pycharm所需环境”后再试一下。退出python按ctrl+z即可。
4、配置PyCharm所需环境
这一步的目的其实和第一步相同,进一步安装所需的依赖库。在caffe根目录的python文件夹下,有一个requirements.txt的清单文件,上面列出了需要的依赖库,按照这个清单安装就可以了。在安装scipy库的时候,需要fortran编译器(gfortran),如果没有这个编译器就会报错,因此,我们可以先安装一下。总的步骤为,进入 caffe/python 目录下,执行安装代码:
sudo apt-get install gfortran
for req in $(cat requirements.txt); do sudo pip install $req; done
安装完成以后执行:
sudo pip install -r requirements.txt
就会看到,安装成功的,都会显示Requirement already satisfied, 没有安装成功的,会继续安装。
5、安装PyCharm
安装过程参考:http://blog.youkuaiyun.com/zhuanshu666/article/details/73554885
专业版的可能无法激活,可以下载免费的社区版。
打开PyCharm客户端的方法:
cd /home/ccem/Downloads/pycharm-2017.3.3/bin #此处按照自己的pycharm解压文件夹的路径修改
sh ./pycharm.sh
6、设置PyCharm解释器
由于系统中安装了多个python版本,需要指定使用哪一个版本。先创建一个工程(命名为test)和一个python文件,创建方法和visual studio类似。然后设置解释器:依次点击:左上角file -- settings..-- Project:test-- Project interpreter --弹出窗口右上角齿轮--add local--System Interpreter --下拉菜单选择 usr/bin/python 2.7 -- OK -- Apply
随后在python文件里输入 import caffe ,点击上方菜单栏里的Run下面的Run运行,显示Process finished with exit code 0 说明运行成功。假如显示找不到某个模块就通过上述步骤在下拉里选择其他解释器进行尝试。选择解释器时下面会列出已经安装的依赖库,安装依赖库最多的那个一般就是要选的那个。
接下来就可以愉快的开始caffe的编程之旅啦!