src/nni_manager/training_service/local/localTrainingServiceForGPU.ts · 892c9c4d7f7803d1e68173ba1d06ebba7d5219bc · OpenDAS / nni

Yan Ni authored Nov 22, 2018

* add gpuNum check for local TS

* set CUDA_VISIBLE_DEVICES to empty string when gpuNum is 0

* remove redundency code

1df750e2

localTrainingServiceForGPU.ts 5.3 KB