mirror of
https://github.com/huggingface/pytorch-image-models.git
synced 2025-06-03 15:01:08 +08:00
See if we can use tcmalloc in test runner
This commit is contained in:
parent
8bf63b6c6c
commit
34522097b1
7
.github/workflows/tests.yml
vendored
7
.github/workflows/tests.yml
vendored
@ -36,11 +36,16 @@ jobs:
|
||||
run: pip install --no-cache-dir torch==${{ matrix.torch }} torchvision==${{ matrix.torchvision }}
|
||||
- name: Install torch on ubuntu
|
||||
if: startsWith(matrix.os, 'ubuntu')
|
||||
run: pip install --no-cache-dir torch==${{ matrix.torch }}+cpu torchvision==${{ matrix.torchvision }}+cpu -f https://download.pytorch.org/whl/torch_stable.html
|
||||
run: |
|
||||
pip install --no-cache-dir torch==${{ matrix.torch }}+cpu torchvision==${{ matrix.torchvision }}+cpu -f https://download.pytorch.org/whl/torch_stable.html
|
||||
sudo apt update
|
||||
sudo apt install -y google-perftools
|
||||
- name: Install requirements
|
||||
run: |
|
||||
if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
|
||||
pip install --no-cache-dir git+https://github.com/mapillary/inplace_abn.git@v1.0.12
|
||||
- name: Run tests
|
||||
env:
|
||||
LD_PRELOAD: /usr/lib/x86_64-linux-gnu/libtcmalloc.so.4
|
||||
run: |
|
||||
pytest -vv --durations=0 ./tests
|
||||
|
Loading…
x
Reference in New Issue
Block a user