使用cmake编译caffe, 使用多GPU，更改nccl无效，

USE_NCCL = 1 
   

    
1
   
   

    
1
   
   

在caffe_root目录中执行命令

    

     

      

       

      
      

       

        $mkdir build
       
      
     

      

       

      
      

       

        $cd build
       
      
     

      

       

      
      

       

        $cmake ..
       
      
     
   

    
1
    
2
    
3
   
   

    
1
    
2
    
3
   
   

显示配置为： 
 
USE_NCCL 为OFF,再使用cmake编译之后使用多gpu进行训练时会报错，提示未使用nccl。

解决方法先看nccl是否安装成功， 
安装步骤为

    

     

      

       

      
      

       

        $ git clone https://github.com/NVIDIA/nccl.git
       
      
     

      

       

      
      

       

        $ cd nccl
       
      
     

      

       

      
      

       

        $ sudo make install -j8
       
      
     
   

    
1
    
2
    
3
   
   

    
1
    
2
    
3
   
   

 
头文件和链接库都已经生成，说明是nccl已经安装好了。

再在CMakeLists.txt里面发现： 
 
在Makefile.config中更改USE_NCCL 后，CMakeLists.txt中的配置是没有发生改变的，手动将OFF改为ON以后，保存再使用cmake编译caffe。 
完成后使用多gpu进行训练。 
 
使用多GPU成功。