CloudWatch metrics for GPU on EC2 G3 instance

gpu_stats.sh

#!/bin/bash

IFS=', ' read -r -a STATS <<< `nvidia-smi --query-gpu=temperature.gpu,memory.used,memory.free,utilization.gpu,utilization.memory --format=csv,nounits,noheader`
#echo ${STATS[@]}

# Send values to CloudWatch
aws cloudwatch put-metric-data --metric-name Temperature --namespace GPUStats --value ${STATS[0]}
aws cloudwatch put-metric-data --metric-name MemoryUsed --namespace GPUStats --unit Megabytes --value ${STATS[1]}
aws cloudwatch put-metric-data --metric-name MemoryFree --namespace GPUStats --unit Megabytes --value ${STATS[2]}
aws cloudwatch put-metric-data --metric-name GPUUtilization --namespace GPUStats --value ${STATS[3]}
aws cloudwatch put-metric-data --metric-name MemoryUtilization --namespace GPUStats --value ${STATS[4]}