国产计算框架mindspore在gpu环境下1.3.0版本的分布式计算组件安装—— (openmpi 和 nccl 的安装,配置,示例代码的运行)
时间:2022-05-11 13:04
前文已经给出1.3.0gpu版本的编译及安装,本文在此基础上进行分布式组件的安装,前文信息参看:
===============================================================
OpenMPI-4.0.3
:MindSpore采用的多进程通信库。
OpenMPI-4.0.3源码下载地址:
参考OpenMPI官网教程安装:。
先安装依赖环境:
sudo apt-get install libibverbs1
解压:
tar -zxvf openmpi-4.0.3.tar.gz
配置:
./configure --prefix=/usr/local --enable-mpi-threads --enable-progress-threads --enable-shared --with-threads=poxis --enable-mpi-thread-multiple --with-cuda=/usr/local/cuda-11.1
(注意:这里我们假设cuda的安装目录为: /usr/local/cuda-11.1 , 该地址可以按照具体情况进行修改)
编译并安装:
make&&sudo make install