You are viewing a plain text version of this content. The canonical link for it is here.

Posted to dev@madlib.apache.org by GitBox <gi...@apache.org> on 2019/04/30 00:05:47 UTC

[GitHub] [madlib] kaknikhil opened a new pull request #379: DL: Replace use_gpu flag with gpus_per_host integer

kaknikhil opened a new pull request #379:  DL: Replace use_gpu flag with gpus_per_host integer 
URL: https://github.com/apache/madlib/pull/379
 
 
   JIRA: MADLIB-1308
   
   Previously, gpus_per_host were hard coded to 4. This commit removes this
   hard coding and takes in this value from the user.
   
   We also tried to use the tensorflow function `list_local_devices` to get
   the count of gpus per host. This did give us the count but would hang
   forever on some segments. So we decided to not use this function.
   
   We now cache the CUDA_VISIBLE_DEVICES env variable (which is set to
   -1 for master) and then reset it at the end of fit function.
   
   Also fixed a bug  in get device name function for postgres
   

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
 
For queries about this service, please contact Infrastructure at:
users@infra.apache.org


With regards,
Apache Git Services