ci(env): add nccl env · 3b6285d7ba - MegEngine - OpenI

ci(env): add nccl env

release-0.6

luzzyzhang

Xinran Xu 5 years ago

parent

80ecabe8c6

commit

3b6285d7ba

1 changed files with 2 additions and 0 deletions

Unified View

							
								+ 2
								
								- 0
							
.github/workflows/ci-cuda.ymlView File
								
																@@ -13,6 +13,8 @@ jobs:
															
																      volumes:
																
																      volumes:
															
																        - /usr/local/cuda-10.1-libs:/usr/local/cuda-10.1-libs
																
																        - /usr/local/cuda-10.1-libs:/usr/local/cuda-10.1-libs
															
																      options: --gpus all --shm-size 1g
																
																      options: --gpus all --shm-size 1g
															
																      env:
															
																        NCCL_LAUNCH_MODE: PARALLEL
															
																    steps:
																
																    steps:
															
																      - name: Checkout MegEngine
																
																      - name: Checkout MegEngine