Caffe 之NCCL安装

最新推荐文章于 2024-06-28 09:34:14 发布

Jancy1072

最新推荐文章于 2024-06-28 09:34:14 发布

阅读量5k

点赞数

CC 4.0 BY-SA版权

分类专栏： AI 文章标签： nvidia gpu

本文链接：https://blog.youkuaiyun.com/shelly1072/article/details/79807831

AI 专栏收录该内容

6 篇文章

订阅专栏

本文介绍了如何在多个GPU上运行Caffe深度学习框架的方法。首先需要安装NVIDIANCCL库来支持多GPU并行计算。安装过程包括克隆NCCL仓库、编译和安装。若遇到编译错误，可在Makefile中指定CXX为gcc-4.8来解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在多个 GPU 上运行 Caffe 需要使用 NVIDIA NCCL。可使用以下命令安装 NCCL：
git clone https://github.com/NVIDIA/nccl.git
cd nccl
sudo make install -j4
NCCL 库和文件头将安装在 /usr/local/lib 和 /usr/local/include 中。

如果遇到报错如下：
Compiling src/all_gather.cu > /home/abc/nccl/build/obj/all_gather.o Compiling src/all_reduce.cu > /home/abc/nccl/build/obj/all_reduce.o src/core.h:35:16: error: expected unqualified-id before ‘long’ src/core.h:35:16: error: expected ‘)’ before ‘long’ src/core.h:99:1: error: ‘DevRing’ does not name a type make: *** [/home/liujing/nccl/build/obj/libwrap.o] Error 1 make: *** Waiting for unfinished jobs.... src/core.h:35:16: error: expected unqualified-id before ‘long’ src/core.h:35:16: error: expected ‘)’ before ‘long’ src/core.h:99:1: error: ‘DevRing’ does not name a type
解决：只要在Makefile文件中指定一下CXX就可以了。如下：
BUILDDIR ?= build BUILDDIR := $(abspath $(BUILDDIR)) CXX := gcc-4.8　　//添加这一行