From 05bc548d37b7a0fd5e75190631843fa98efba816 Mon Sep 17 00:00:00 2001 From: Hui Zhang Date: Fri, 5 Mar 2021 06:26:38 +0000 Subject: [PATCH] fix libri train.sh --- examples/librispeech/local/train.sh | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/librispeech/local/train.sh b/examples/librispeech/local/train.sh index 59a94181b..507947e9e 100644 --- a/examples/librispeech/local/train.sh +++ b/examples/librispeech/local/train.sh @@ -1,6 +1,8 @@ #! /usr/bin/env bash export FLAGS_sync_nccl_allreduce=0 +# https://github.com/PaddlePaddle/Paddle/pull/28484 +export NCCL_SHM_DISABLE=1 ngpu=$(echo ${CUDA_VISIBLE_DEVICES} | python -c 'import sys; a = sys.stdin.read(); print(len(a.split(",")));') echo "using $ngpu gpus..."