目前存在的问题:
$ python
>>import caffe
会报错:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: No module named caffe
查询得到 >> 这可能是因为没有把caffe中的python导入到解释器中
在bashrc中添加 添加:
export PYTHONPATH=/这里更替为你自己caffe目录下的python文件夹的路径/caffe/python:$PYTHONPATH
但是还不行
搞了一天,下次再弄吧。。
我的安装环境:
- OS:Ubuntu 18.04 64 bit
- 显卡:NVidia GTX 1650
- CUDA:10.0
- cuDNN:7.6.0
- python:2.7(默认)/ 3.6
- OpenCV:2.4.11 / 3.2.0 (默认,可通过
pkg-config --modversion opencv
查看 )/ 4.1.2/4.4.0
第一步:下载源码
在所要安装的路径下:
git clone https://github.com/BVLC/caffe.git
直接下载,复制过来也ok…
编译
我的是直接下载的,所以需要解压,然后进入解压之后的目录,进行下面的操作
1. 将Makefile.config.example
文件复制一份并更名为 Makefile.config
:
sudo cp Makefile.config.example Makefile.config
复制一份的原因是编译 caffe 时需要的是 Makefile.config 文件,而Makefile.config.example 只是caffe 给出的配置文件例子,不能用来编译 caffe。
2. 文件修改
2.1 修改Makefile.config文件
(1)应用 cudnn
#USE_CUDNN := 1
修改为
USE_CUDNN := 1
(2)选择opencv (默认使用opencv2,但是我是opencv3的,所以需要修改一下)
#OPENCV_VERSION := 3
修改为:
OPENCV_VERSION := 3
(3)根据自身cuda版本修改(这里借用一下CAU_Ayao博主的图),文件中都有提示(蓝字部分)
CUDA_ARCH := #-gencode arch=compute_20,code=sm_20 \
#-gencode arch=compute_20,code=sm_21
-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 \