abhishekkrthakur/tez

Can I set the device id of GPUs?

Caizifen opened this issue · 0 comments

Caizifen commented

I want to train the model using gpus, so can I set the id of device? Since I have multiple gpus on my computer.