Why are tests running on cpu?

[jiaqiw09]

I just want to do the whole tests to see what this package do. However I find all models is running on cpu (test_models.py) and performance is so slow?

[rwightman]

@jiaqiw09 because github runners are CPU

TORCH_DEVICE=cuda pytest -vv tests/

But it's still going to be pretty slow as there are lots of tests...

[jiaqiw09]

thanks. I just test it and find it's very slow. I am trying to running it on mutiple gpus, I hope it will work

Best