Nandish Jayaram created MADLIB-1308: ---------------------------------------
Summary: Change GPU related hardcoded things in madlib_keras.py_in Key: MADLIB-1308 URL: https://issues.apache.org/jira/browse/MADLIB-1308 Project: Apache MADlib Issue Type: Improvement Components: Deep Learning Reporter: Nandish Jayaram Fix For: v1.16 Based on the code in PR [https://github.com/apache/madlib/pull/355:] # Currently in madlib_keras.py_in , we hardcod the following things ## gpus_per_host = 4 ## device_name = '/cpu:0' or '/gpu:0' ( can the device ever be not named gpu0 or cpu0 ? ) # Look into and document the usage of {{CUDA_VISIBLE_DEVICES}} when gpu_only is set to TRUE. Currently we set it to str(current_seg_id % gpus_per_host). How does this logic work and will it always work? How would this logic change for Postgres, since it has no segment_id. # What happens if no of gpus < no of segments no of gpus > no of segments -- This message was sent by Atlassian JIRA (v7.6.3#76005)