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)
