centos7 安装nccl
时间: 2023-11-01 13:08:59 浏览: 357
Centos7下安装MongoDB
安装NCCL的步骤如下:
1. 首先,你需要安装git和cmake。你可以使用以下命令安装它们:
```
sudo yum install git
sudo yum install cmake
```
2. 克隆NCCL的git仓库:
```
git clone https://github.com/NVIDIA/nccl.git
cd nccl
```
3. 编译和安装NCCL:
```
make -j4
sudo make install
```
4. 配置NCCL:
打开caffe/cmake/Modules/FindNCCL.cmake文件,将以下两行代码替换为你的NCCL路径:
```
set(NCCL_INC_PATHS /home/wangfeng/packages/nccl/include )
set(NCCL_LIB_PATHS /home/wangfeng/packages/nccl/lib )
```
5. 重新编译caffe:
```
cd caffe
mkdir build
cd build
cmake ..
make -j4
```
阅读全文