search

microsoft / tensorflow-directml

Fork of TensorFlow accelerated by DirectML
Apache License 2.0
454 stars 32 forks source link

how to set the default device using cAPI? #401

Open argman opened 1 year ago

argman commented 1 year ago

if I have multi gpu cards, how to set the default device, when using cuda, we can set CUDA_VISIBLE_DEVICES, how to do this in directml?