-
|
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? |
Beta Was this translation helpful? Give feedback.
Answered by
rwightman
Apr 21, 2024
Replies: 1 comment 1 reply
-
|
@jiaqiw09 because github runners are CPU
But it's still going to be pretty slow as there are lots of tests... |
Beta Was this translation helpful? Give feedback.
1 reply
-
|
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 |
Beta Was this translation helpful? Give feedback.
Answer selected by
jiaqiw09
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
@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...