Thomas Graves created SPARK-27725:
-------------------------------------
Summary: GPU Scheduling - add an example discovery Script
Key: SPARK-27725
URL: https://issues.apache.org/jira/browse/SPARK-27725
Project: Spark
Issue Type: Story
Components: Examples
Affects Versions: 3.0.0
Reporter: Thomas Graves
We should add an example script that can be used to discovery GPU's and output
the correctly formatted JSON.
Something like below, but it needs to be tested on various systems with more
then 2 gpu's:
ADDRS=`nvidia-smi --query-gpu=index --format=csv,noheader | sed 'N;s/\n/\",\"/'`
COUNT=`echo $ADDRS | tr -cd , | wc -c`
ALLCOUNT=`expr $COUNT + 1`
#echo \{\"name\": \"gpu\", \"addresses\":[\"$ADDRS\"]}
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]