Skip to content

GitLab

Unverified Commit b4dee63c authored by Zaida Zhou's avatar Zaida Zhou Committed by GitHub
Browse files

[FIx] Build CUDA ops in CI (#2794)

parent 387b63b3
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 0 deletions
.github/workflows/merge_stage_test.yml
View file @ b4dee63c
......@@ -175,6 +175,9 @@ jobs:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel
env:
FORCE_CUDA: 1
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
strategy:
matrix:
python-version: [3.7]
......@@ -209,6 +212,9 @@ jobs:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
env:
FORCE_CUDA: 1
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
strategy:
matrix:
python-version: [3.7]
......@@ -243,6 +249,9 @@ jobs:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:1.13.0-cuda11.6-cudnn8-devel
env:
FORCE_CUDA: 1
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
strategy:
matrix:
python-version: [3.7]
......
.github/workflows/pr_stage_test.yml
View file @ b4dee63c
......@@ -23,6 +23,9 @@ jobs:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:1.8.1-cuda10.2-cudnn7-devel
env:
FORCE_CUDA: 1
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
strategy:
matrix:
python-version: [3.7]
......@@ -65,6 +68,9 @@ jobs:
runs-on: ubuntu-22.04
container:
image: pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
env:
FORCE_CUDA: 1
MMCV_CUDA_ARGS: -gencode=arch=compute_61,code=sm_61
strategy:
matrix:
python-version: [3.7]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment