mirror of https://github.com/open-mmlab/mmocr.git
parent
2341bc5528
commit
ee1dbf1037
|
@ -135,10 +135,7 @@ 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 .
|
||||
TORCH_CUDA_ARCH_LIST=7.0 pip install -e .
|
||||
- name: Run unittests and generate coverage report
|
||||
run: |
|
||||
coverage run --branch --source mmocr -m pytest tests/
|
||||
|
|
1
setup.py
1
setup.py
|
@ -102,7 +102,6 @@ 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