pip install pytest-instafail | |
pytest --instafail | |
To GPU or not to GPU | |
On a GPU-enabled setup, to test in CPU-only mode add CUDA_VISIBLE_DEVICES="": | |
CUDA_VISIBLE_DEVICES="" pytest tests/utils/test_logging.py | |
or if you have multiple gpus, you can specify which one is to be used by pytest. |