Remove GPU check
proteneer opened this issue · comments
Yutong Zhao commented
the GPU check is causing us to allocate all the GPUs visible via CUDA_VISIBLE_DEVICES. This causes issues if we could see N gpus but only specify using K (passed into --gpus). The check causes us to grab all N gpus as opposed to K gpu slots.
Yutong Zhao commented
this is more likely to be an older tf bug. closing