ubuntu 18.04 中 编译 FasterTransformer,与缺少安装包

前提:A100, cuda 11.6  cudnn8 nccl zlib1g-dev

git clone  --recursive  https://github.com/NVIDIA/FasterTransformer.git
git submodule update --recursive --init

#保险起见
cd FasterTransformer
mkdir build
cd build
cmake ..
make -j

执行示例:

BF16数据格式:

 

错误处理:

如遇到如下错误,可如下方法解决:

fatal error: nccl.h: No such file or directory

fatal error: zlib.h: No such file or direct

安装 nccl gpu 通信包

GitHub - NVIDIA/nccl: Optimized primitives for collective multi-GPU communication

如下都是官方步骤:

下载源代码:

​
git clone --recursive https://github.com/NVIDIA/nccl.git

​

 编译源代码:

$ cd nccl
$ make -j src.build

deb包:

$ # Install tools to create debian packages
$ sudo apt install build-essential devscripts debhelper fakeroot
$ # Build NCCL deb package
$ make pkg.debian.build
$ ls build/pkg/deb/

tar包:

$ make pkg.txz.build
$ ls build/pkg/txz/

安装:

sudo dpkg -i libnccl-dev_2.13.4-1+cuda11.6_amd64.deb
sudo dpkg -i libnccl2_2.13.4-1+cuda11.6_amd64.deb

测试是否安装成功:

$ git clone https://github.com/NVIDIA/nccl-tests.git
$ cd nccl-tests
$ make
$ ./build/all_reduce_perf -b 8 -e 256M -f 2 -g <ngpus>

顺带安装一下zlib:

$ sudo apt-get install zlib1g-dev

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值