caffe没有提供多个GPU培训

Question

当尝试在多个gpus中使用caffe时，我遇到了一些问题。在执行以下命令时，我将得到下面的错误日志显示：

caffe train -solver $SOLVER -gpu 0,1 2>&1 | tee $LOGGING

F0409 14:17:22.355074 12079 caffe.cpp:254] Multi-GPU execution not available - rebuild with USE_NCCL
*** Check failure stack trace: ***
    @     0x2aee66002b2d  google::LogMessage::Fail()
    @     0x2aee66004995  google::LogMessage::SendToLog()
    @     0x2aee660026a9  google::LogMessage::Flush()
    @     0x2aee6600542e  google::LogMessageFatal::~LogMessageFatal()
    @           0x40c172  train()
    @           0x4084f3  main
    @     0x2aee78f67b35  __libc_start_main
    @           0x408f0b  (unknown)

有人能解释一下这是怎么回事吗？有什么我不知道的臭虫吗？

Answer 1

1. 安装CUDA
2. 安装cuDNN
3. 安装依赖项$ sudo apt-get Install libprotobuf dev libleveldb-dev libsnappy-dev libopencv-dev libhdf5 5-serial dev protobuf编译程序libg资-dev libgoogle-glog-dev liblmdb-dev libatlas-base-dev git $ sudo -get-no- install -推荐libboost all-dev-dev
4. 安装NCCL NVIDIA NCCL需要在多个GPU上运行Caffe。可以使用以下命令安装NCCL： $ git克隆https://github.com/NVIDIA/nccl.git $ cd nccl $ sudo make -j NCCL库和头将安装在/usr/local/lib和/usr/local/include中。
5. 安装Caffe 取消USE_CUDNN := 1行的注释。这使cuDNN加速。取消USE_NCCL := 1行的注释。这使得NCCL能够在多个GPU上运行Caffe。 保存并关闭文件。您现在已经准备好编译Caffe了。 $ make all -j 当此命令完成后，Caffe二进制文件将在build/tools/caffe上可用。