GPU上使用容器+mpirun进行分布式训练

此贴是用户的一个提问的经验帖子，在此做下案例分析。帖子来源见： 华为云论坛_云计算论坛_开发者论坛_技术论坛-华为云

问题

1、在容器中启动训练的时候，如何指定通信端口？ 2.如何进入到容器中并且在进入一个conda环境中呢，应该如何操作？

回复

问题1：

mpirun 可以通过 -mca 参数中传入容器的端口号 port，具体如下图所示 

 注意：容器的端口号必须和主机共用端口号，即在docker run 后面加上--network host参数，

然后将 /etc/.ssh/sshd_config中的port修改为自己想设置的端口号（必须修改，不能是默认的22，不然和会和主机端口号冲突）

问题2：

进入conda环境的话可以通过修改 容器中的~/.bashrc中加入 conda activate