Tensorflow on shared GPUs: how to automatically select the one that is unused

前端 未结 2 1632
情话喂你
情话喂你 2020-12-17 19:24

I have access through ssh to a cluster of n GPUs. Tensorflow automatically gave them names gpu:0,...,gpu:(n-1).

Others have access too and sometimes they take random

2条回答
  •  误落风尘
    2020-12-17 19:46

    An implementation along the lines of Yaroslav Bulatov's solution is available on https://github.com/bamos/setGPU.

提交回复
热议问题