mirror of https://github.com/open-mmlab/mmocr.git
parent
49ffc1b64f
commit
e7a888d872
|
@ -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: |
|
||||
|
|
2
setup.py
2
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'))
|
||||
|
|
Loading…
Reference in New Issue