目录:
Ubuntu14.04 安装Caffe(仅CPU)
Ubuntu14.04 安装Cuda
Ubuntu14.04 CuDNN安装(Caffe + Cuda7.0下)
重新编译:
使用GPU
- 见 Ubuntu14.04安装Caffe(仅CPU)#编译caffe
- 修改Makefile.config文件:加上CPU_ONLY:=1的注释,重新编译
查看CUDA计算容量:
$ ./~/cuda/NVIDIA_CUDA-7.0_Samples/bin/x86_64/linux/release/deviceQuery
查看CUDA Capability Major/Minor version number这一项,为5.2
在Makefile.config文件中,修改
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_50,code=compute_50
加入倒数第二行那一句话
重新配置pycaffe和matcaffe:
即可