Replicating the function of tf.config.list_physical_devices

[Tom B]

Hello SIG-members,

is there a way to access the list of physical devices provided through the function mentioned in the title in Java? list_physical_devices

The use-case I'm currently trying to solve:
- I want the user of my application to be able to select the device(s) on which Tensorflow is going to run

- (optional) I also want to check the availability of GPUs, to ensure that CUDA is correctly installed. I reason that they should not be available to Tensorflow if anything is missing, correct me if I am wrong on that.

[Karl Lessard]

Hi Tom,

There is an internal endpoint that should return you the same list as in Python. There are a few other endpoints in that same file related to devices that you can take a look at.

These endpoints though are not meant to be used directly by the end-users and should be wrapped instead within a Java-friendly API. The wrapper for such an endpoint has not been added yet. If they work for you, would you be interested to make that contribution? (maybe something like `List<DeviceSpec> TensorFlow.listDevices(DeviceType... deviceType)` or any better suggestion?)

Thanks,

Karl

--
You received this message because you are subscribed to the Google Groups "SIG JVM" group.
To unsubscribe from this group and stop receiving emails from it, send an email to jvm+uns...@tensorflow.org.
To view this discussion on the web visit https://groups.google.com/a/tensorflow.org/d/msgid/jvm/067d8f39-060d-41d9-a485-1fb1db913060n%40tensorflow.org.

[Tom B]

Thanks for the pointers Karl! I'll gladly make that contribution as soon as I find a good timeslot to get to it, as this topic is currently further down the priority list for me. No promises, but I think I can go back to it at the start of december.

Best,

Tom

[Tom B]

Hi everyone,

I'm having a hard time trying to unterstand how the wrapping and converting of objects works, especially in regards to converting from TF_DeviceList to List<DeviceSpec>.

Is anyone willing to give me some pointers or has time for a short bit of partner-coding to get me started with the project?

Best,

Tom

[Samuel Audet]

Hi, Tom,

TFE_ContextListDevices() appears to get called only in this function from Python:

https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/eager/context.py#L477

I think we basically just need to port that function to Java :)

There's also code in Go and C++ calling it:

https://github.com/tensorflow/tensorflow/search?q=TFE_ContextListDevices

If there is anything unclear, please let me know!

Samuel