Skip to content

Commit 30a0986

Browse files
Fix
1 parent 6a4a350 commit 30a0986

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

ext/TerminalSystemMonitorCUDAExt/TerminalSystemMonitorCUDAExt.jl

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,18 +46,19 @@ using TerminalSystemMonitor: extract_number_and_unit
4646
└41s────────────────30s──────────────────20s─────────────────10s─────────────────0s┘
4747
"""
4848

49-
function _plot_gpu_utilization_rates(dev::CUDA.CuDevice)
49+
function _plot_gpu_utilization_rates(gpu_id, dev::CUDA.CuDevice)
5050
mig = uuid(dev) != parent_uuid(dev)
5151
nvml_dev = CUDA.NVML.Device(uuid(dev); mig)
52-
x = CUDA.NVML.name(nvml_dev)
52+
x = string(gpu_id)
5353
y = 100 * CUDA.NVML.utilization_rates(nvml_dev).compute
5454
return barplot([x], [y], maximum = 100, width = max(5, 15))
5555
end
5656

5757
function TerminalSystemMonitor.plot_gpu_utilization_rates(::Type{CUDADevice})
5858
plts = []
59-
for dev in CUDA.devices()
60-
push!(plts, _plot_gpu_utilization_rates(dev))
59+
for (jli, dev) in enumerate(CUDA.devices())
60+
gpu_id = jli - 1 # from 1-based to 0-based index
61+
push!(plts, _plot_gpu_utilization_rates(gpu_id, dev))
6162
end
6263
return plts
6364
end

src/TerminalSystemMonitor.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,7 @@ function main(dummyargs...)
140140

141141
f /= prod(UnicodePlots.panel.(plot_cpu_memory_utilization()))
142142

143-
if isdefined(Main, :CUDA)
143+
if isdefined(Main, :CUDA) && getproperty(getproperty(Main, :CUDA), :functional)()
144144
cudaplts = []
145145
n = max(1, cols ÷ 50)
146146
plts1::Vector{Any} = plot_gpu_utilization_rates(MLDataDevices.CUDADevice)

0 commit comments

Comments
 (0)