From 8fcc072788689871a033ab9d3411e4e9d0ef1f43 Mon Sep 17 00:00:00 2001 From: luzzyzhang Date: Wed, 22 Apr 2020 15:28:36 +0800 Subject: [PATCH] ci(workflow): add gpu ci workflow --- .github/workflows/ci-cuda.yml | 29 +++++++++++++++++++++++++++++ ci/docker_env/Dockerfile | 5 +++-- 2 files changed, 32 insertions(+), 2 deletions(-) create mode 100644 .github/workflows/ci-cuda.yml diff --git a/.github/workflows/ci-cuda.yml b/.github/workflows/ci-cuda.yml new file mode 100644 index 00000000..2fac75c9 --- /dev/null +++ b/.github/workflows/ci-cuda.yml @@ -0,0 +1,29 @@ +name: CI GPU + +on: + push: + branches: [master] + pull_request: + +jobs: + gpu-test: + runs-on: self-hosted + container: + image: localhost:5000/megengine-ci:latest + volumes: + - /usr/local/cuda-10.1-libs:/usr/local/cuda-10.1-libs + options: --gpus all + + steps: + - name: Checkout MegEngine + uses: actions/checkout@v2 + - name: Checkout submodules + run: | + ./third_party/prepare.sh + ./third_party/install-mkl.sh + - name: Build MegEngine + run: ./ci/cmake.sh cuda + - name: Python test + run: ./ci/run_python_test.sh + - name: C++ test + run: ./ci/run_cpp_test.sh cuda diff --git a/ci/docker_env/Dockerfile b/ci/docker_env/Dockerfile index 8f525f5c..4496fc99 100644 --- a/ci/docker_env/Dockerfile +++ b/ci/docker_env/Dockerfile @@ -35,5 +35,6 @@ RUN git lfs install ENV PATH=${PATH}:/usr/local/cuda/bin \ LIBRARY_PATH=${LIBRARY_PATH}:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda/lib/stubs:/usr/local/cuda/lib64/stubs:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib64:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib64 \ - LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda/lib/stubs:/usr/local/cuda/lib64/stubs:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib64:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib64:/usr/local/cuda/lib64/stubs/libcuda.so:/tmp/build/cuda/dnn/cuda-stub/libcuda.so.1 \ - CPATH=${CPATH}:/usr/local/cuda/include:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/include:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/include \ No newline at end of file + LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/lib64:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/lib64:/tmp/build/cuda/dnn/cuda-stub/libcuda.so \ + CPATH=${CPATH}:/usr/local/cuda/include:/usr/local/cuda-10.1-libs/cudnn-v7.6.0/include:/usr/local/cuda-10.1-libs/TensorRT-5.1.5.0/include \ + CUDA_BIN_PATH=/usr/local/cuda -- GitLab