Ubuntu14.04 安装Caffe(GPU)

目录:

  • Ubuntu1404 安装Caffe仅CPU
  • Ubuntu1404 安装Cuda
  • Ubuntu1404 CuDNN安装Caffe Cuda70下
  • 重新编译
    • 使用GPU
    • 查看CUDA计算容量
  • 重新配置pycaffe和matcaffe

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:

  • Ubuntu14.04 安装Caffe(仅CPU)#配置pycaffe
  • Ubuntu14.04 安装Caffe(仅CPU)#配置matcaffe

即可

你可能感兴趣的:(ubuntu,caffe)