Browse Source

ci(env): add nccl env

release-0.6
luzzyzhang Xinran Xu 5 years ago
parent
commit
3b6285d7ba
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      .github/workflows/ci-cuda.yml

+ 2
- 0
.github/workflows/ci-cuda.yml View File

@@ -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


Loading…
Cancel
Save