nvidia_vram_usage_by_user.sh

nvidia_vram_usage_by_user.sh

#!/bin/bash

# Shows which processes & users are consuming GPU memory for Nvidia GPUs.

echo "User, PID, Used Memory, GPU ID"

# Get the GPU processes information
nvidia-smi --query-compute-apps=pid,used_memory,gpu_bus_id --format=csv,noheader | while read line
do
  # Extract PID from the line - use cut instead of awk for CSV format
  pid=$(echo $line | cut -d',' -f1 | tr -d ' ')
  
  # Get username for the process
  username=$(ps -o user= -p $pid 2>/dev/null)
  
  # Check if we got a username and the process exists
  if [ ! -z "$username" ]; then
    echo "$username, $line"
  fi
done