如果要使用多GPU 训练,需要安装nccl
首先我们需要在Makefile.config 里面加入:USE_NCCL=1
先make all-j8 看caffe 编译会不会报错
如果报错
然后安装 nccl
nccl 的安装方法为:
git clone https://github.com/NVIDIA/nccl.git
cd nccl
sudo make install -j8 如果你是在服务器上安装且没有管理员权限
那么你需要修改:
改成自己目录下的local 就可以了 make install -j8
因为你自己安装caffe 的时候会自己在本机建一个属于自己的local 文件的位置
把这个local 的位置加入到./bashrc 下面
至此 nccl 安装完毕
然后返回到caffe 下面 编译caffe 就ok