mxnet-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] marcoabreu commented on issue #10354: Expose the number of GPUs.
Date Sun, 01 Apr 2018 00:29:23 GMT
marcoabreu commented on issue #10354: Expose the number of GPUs.
URL: https://github.com/apache/incubator-mxnet/pull/10354#issuecomment-377733428
 
 
   Hey Tobias, definitely helpful - I've seen myself requiring that information quite a few
times :)
   
   I'm currently thinking about a way how to test this properly, but the only way I can think
of is calling nvidia-smi and that's pretty ugly... I think we could go with a test in https://github.com/apache/incubator-mxnet/tree/master/tests/python/gpu
which checks for ```num_gpu > 0``` and leave out CPU tests as we don't have cpu-only tests
so far. What do you think?

----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on 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

Mime
View raw message