How many GPUs are available:

thon
from transformers.testing_utils import get_gpu_count
n_gpu = get_gpu_count()  # works with torch and tf

Testing with a specific PyTorch backend or device
To run the test suite on a specific torch device add TRANSFORMERS_TEST_DEVICE="$device" where $device is the target backend. For example, to test on CPU only: