caffe下多gpu训练,NCCL的配置

来源:互联网 发布:桌面公告通知软件 编辑:程序博客网 时间:2024/06/07 06:29

一、caffe下gpu训练,环境配置——安装NCCL和修改Makefile.config文件

  • 安装NCCL1.0(先安装cuda和cudnn)
    • 下载NCCL1.0(cuda版本要求:cuda version>6.0,gcc 版本>=gcc 4.8)
      NCCL下载地址
    • 安装NCCL
 cd nccl-mastersudo make CUDA_HOME=/usr/local/cuda-8.0/ test #根据自己cuda路径修改CUDA_HOMEsudo make installsudo ldconfig 
  • 修改caffe目录下的Makefile.config
#USE_NCCL:=1 

改为

USE_NCCL:=1 #把前面的#注释掉
  • 重新编译caffe
make all -j8make pycaffe -j8make testmake runtest -j8 #如果报错在前面加sudomake install
原创粉丝点击