How to assign a specific GPU when auto-tuning?

Questions
1

Hi,
I am trying to run the auto-tuning program on a specific GPU. But I don’t find a proper API to set. How can I do it?

2

have you solved this problem?

3

@ comaniac any advice? Thank you!

4

export CUDA_VISIBLE_DEVICES="your gpu id"

1 Like
5

wow thank you! I’ll try it.