search

awslabs / keras-apache-mxnet

[DEPRECATED] Amazon Deep Learning's Keras with Apache MXNet support
https://github.com/awslabs/keras-apache-mxnet/wiki
Other
289 stars 65 forks source link

change how mxnet backend detect gpus #230

Closed roywei closed 4 years ago

roywei commented 5 years ago

Currently the way mxnet backend detect number of gpus does not work on some of the platforms, it relies on running nvidia-smi command line which is not available on all operating systems. ref: https://github.com/awslabs/keras-apache-mxnet/blob/master/keras/backend/mxnet_backend.py#L5578 Need a better way and warning message to automatically detect gpus.

Current work arounds is to manually specify gpus through context on other platforms


model.compile( ... , context=["gpu(0)", "gpu(1)"]
roywei commented 4 years ago

updated via https://github.com/awslabs/keras-apache-mxnet/pull/236/files