Signed-off-by: lizz <lizz@sensetime.com>
pull/43/head
lizz 2021-04-10 16:26:54 +08:00
parent 49ffc1b64f
commit e7a888d872
2 changed files with 4 additions and 1 deletions

View File

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

View File

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