diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index b8437c50..0c817d2c 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -135,6 +135,9 @@ jobs: run: | rm -rf .eggs python setup.py check -m -s + echo $CUDA_HOME + ls /usr/local/ + python -c 'import torch;print(torch.cuda.is_available())' TORCH_CUDA_ARCH_LIST=7.0 pip install . - name: Run unittests and generate coverage report run: | diff --git a/setup.py b/setup.py index 4a0a9bcb..ee9e2d30 100644 --- a/setup.py +++ b/setup.py @@ -102,7 +102,7 @@ def parse_requirements(fname='requirements.txt', with_version=True): def get_rroi_align_extensions(): - + print('XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX') extensions_dir = 'mmocr/models/utils/ops/rroi_align/csrc/csc' main_file = glob.glob(os.path.join(extensions_dir, '*.cpp')) source_cpu = glob.glob(os.path.join(extensions_dir, 'cpu', '*.cpp'))