TensorFlowSharp icon indicating copy to clipboard operation
TensorFlowSharp copied to clipboard

Published 20 hours ago • verified publisher icon migueldeicaza

How do I select different GPU?

Open janipaDev opened this issue 5 years ago • 3 comments

Hi , I have a computer with 4 NVIDIA GPU. I use the tensorflow-batteries-windows-x64-gpu to make tensorflowsharp run on GPU.

However , I had no idea how to select and run on specific GPU.

Anyone has any idea?

janipaDev avatar Dec 09 '19 06:12 janipaDev

Same problem. How can specify "device/GPU:0" during the import of the graph? There isn t documentation or example. Maybe "SessionOptions" and "GPUOptions" but how to use them?

DavideCatto avatar Dec 10 '19 16:12 DavideCatto

You can reference the issue #206: https://github.com/migueldeicaza/TensorFlowSharp/issues/206#issuecomment-359938105. In your case, you should add the GPU number in configuration in python. Good luck!

weiSupreme avatar Dec 17 '19 13:12 weiSupreme

Hi , I have a computer with 4 NVIDIA GPU. I use the tensorflow-batteries-windows-x64-gpu to make tensorflowsharp run on GPU.

However , I had no idea how to select and run on specific GPU.

Anyone has any idea?

tensorflow-batteries-windows-x64-gpu Where can I download it?

bbhxwl avatar Apr 01 '21 04:04 bbhxwl