Skip to content

Commit 4f35cd1

Browse files
committed
issue #25: NCCL_P2P_DISABLE
1 parent afdb4e1 commit 4f35cd1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

training_cli.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99

1010
hps = utils.get_hparams()
1111
os.environ["CUDA_VISIBLE_DEVICES"] = hps.gpus.replace("-", ",")
12+
os.environ["NCCL_P2P_DISABLE"] = 1
1213
from random import shuffle, randint
1314

1415
import torch

0 commit comments

Comments
 (0)