Files
ai-toolkit/ui/src/components/GPUWidget.tsx
T

96 lines
3.5 KiB
TypeScript

import React, { useState } from 'react';
import { GpuInfo } from '@/types';
interface GPUWidgetProps {
gpu: GpuInfo;
}
export default function GPUWidget({ gpu }: GPUWidgetProps) {
// Helper to format memory values
const formatMemory = (mb: number): string => {
if (mb >= 1024) {
return `${(mb / 1024).toFixed(2)} GB`;
}
return `${mb} MB`;
};
// Helper to determine temperature color
const getTemperatureColor = (temp: number): string => {
if (temp < 50) return 'text-green-600';
if (temp < 80) return 'text-yellow-600';
return 'text-red-600';
};
return (
<>
<div
key={gpu.index}
className="bg-gray-900 rounded-lg shadow-lg overflow-hidden hover:shadow-xl transition-shadow duration-300 px-2 py-2"
>
<div className="bg-gray-800 text-white px-2 py-1 flex justify-between items-center">
<h2 className="font-bold text-sm truncate">{gpu.name}</h2>
<span className="text-xs bg-gray-700 rounded px-1 py-0.5">GPU #{gpu.index}</span>
</div>
<div className="p-2">
<div className="mb-2 flex items-center">
<p className="text-xs text-gray-500 mr-1">Temperature:</p>
<p className={`text-sm font-bold ${getTemperatureColor(gpu.temperature)}`}>{gpu.temperature}°C</p>
</div>
<div className="">
<p className="text-xs text-gray-600 mb-0.5">GPU Utilization</p>
<div className="w-full bg-gray-500 rounded-full h-1.5">
<div
className={`h-1.5 rounded-full ${gpu.utilization.gpu < 30 ? 'bg-green-500' : gpu.utilization.gpu < 70 ? 'bg-yellow-500' : 'bg-red-500'}`}
style={{ width: `${gpu.utilization.gpu}%` }}
></div>
</div>
<p className="text-right text-xs mt-0.5">{gpu.utilization.gpu}%</p>
</div>
<div className="mb-2">
<p className="text-xs text-gray-600 mb-0.5">Memory Utilization</p>
<div className="w-full bg-gray-500 rounded-full h-1.5">
<div
className="h-1.5 rounded-full bg-blue-500"
style={{ width: `${(gpu.memory.used / gpu.memory.total) * 100}%` }}
></div>
</div>
<div className="flex justify-between text-xs mt-0.5">
<span>
{formatMemory(gpu.memory.used)} / {formatMemory(gpu.memory.total)}
</span>
<span>{((gpu.memory.used / gpu.memory.total) * 100).toFixed(1)}%</span>
</div>
</div>
<div className="grid grid-cols-2 gap-2 mb-2">
<div>
<p className="text-xs text-gray-500 mb-0.5">Power</p>
<p className="text-sm font-medium">
{gpu.power.draw.toFixed(1)}W / {gpu.power.limit.toFixed(1)}W
</p>
</div>
<div>
<p className="text-xs text-gray-500 mb-0.5">Memory Clock</p>
<p className="text-sm font-medium">{gpu.clocks.memory} MHz</p>
</div>
</div>
<div className="mt-1 pt-1 border-t border-gray-600 grid grid-cols-2 gap-2">
<div>
<p className="text-xs text-gray-500 mb-0.5">Graphics Clock</p>
<p className="text-sm font-medium">{gpu.clocks.graphics} MHz</p>
</div>
<div className="">
<p className="text-xs text-gray-500 mb-0.5">Driver Version</p>
<p className="text-sm font-medium">{gpu.driverVersion}</p>
</div>
</div>
</div>
</div>
</>
);
}