Module: NvidiaSmi

Defined in:
lib/docker_pid/nvidia_smi.rb

Defined Under Namespace

Classes: Process

Class Method Summary collapse

Class Method Details

.index_by_uuid(uuid) ⇒ Object 



27
28
29
30
 
# File 'lib/docker_pid/nvidia_smi.rb', line 27

def self.index_by_uuid(uuid)
  gpu = NvidiaSmi.gpus.find{|g| g.uuid == uuid }
  @gpu_index = gpu.index.to_i
end

.processes ⇒ Object 



17
18
19
20
21
22
23
24
25
 
# File 'lib/docker_pid/nvidia_smi.rb', line 17

def self.processes
  data = `nvidia-smi --query-compute-apps=pid,process_name,used_memory,gpu_uuid --format=csv,nounits,noheader`
  procs = []
  CSV.parse(data) do |row|
    row = row.map{|c| c.strip }
    procs << Process.new(*row)
  end
  procs
end