From 3b6285d7baa3a0095e6ef22c3e59bf379ad80e98 Mon Sep 17 00:00:00 2001
From: luzzyzhang <luzzyzhang@gmail.com>
Date: Tue, 23 Jun 2020 19:13:04 +0800
Subject: [PATCH] ci(env): add nccl env

---
 .github/workflows/ci-cuda.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/ci-cuda.yml b/.github/workflows/ci-cuda.yml
index e1614f0f6..e643bcb32 100644
--- a/.github/workflows/ci-cuda.yml
+++ b/.github/workflows/ci-cuda.yml
@@ -13,6 +13,8 @@ jobs:
       volumes:
         - /usr/local/cuda-10.1-libs:/usr/local/cuda-10.1-libs
       options: --gpus all --shm-size 1g
+      env:
+        NCCL_LAUNCH_MODE: PARALLEL
 
     steps:
       - name: Checkout MegEngine
-- 
GitLab