administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-05-01 19:17:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

docs: improve NVIDIA GPU count input

Browse Source
This commit is contained in:
ZhaiSoul 2026-04-29 14:42:29 +08:00
parent 6cc4db1103
commit 4b421c66a5
3 changed files with 17 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
docs/src/components/DockerComposeGenerator/components/NvidiaGpuConfig.tsx
View File

@ -16,6 +16,8 @@ export default function NvidiaGpuConfig({
onGpuCountChange, onGpuCountChange,
onGpuDeviceIdChange, onGpuDeviceIdChange,
}: Props) { }: Props) {
const showDeviceId = gpuCount !== "";
return ( return (
<div className={styles.nvidiaConfig}> <div className={styles.nvidiaConfig}>
<div className={styles.formGroup}> <div className={styles.formGroup}>
@ -25,16 +27,15 @@ export default function NvidiaGpuConfig({
<input <input
id="dcg-gpu-count" id="dcg-gpu-count"
type="text" type="text"
inputMode="numeric"
pattern="[0-9]*"
className={styles.input} className={styles.input}
value={gpuCount} value={gpuCount}
placeholder="1 or all" placeholder="all"
onChange={(e) => onGpuCountChange(e.target.value)} onChange={(e) => onGpuCountChange(e.target.value.replace(/\D/g, ""))}
/> />
<p className={styles.helpText}>
Enter a number (e.g. 1, 2, 3) or &quot;all&quot; to use all GPUs
</p>
</div> </div>
{gpuCount !== "all" && ( {showDeviceId && (
<div className={styles.formGroup}> <div className={styles.formGroup}>
<label htmlFor="dcg-gpu-device-id" className={styles.label}> <label htmlFor="dcg-gpu-device-id" className={styles.label}>
GPU device IDs (required, comma-separated): GPU device IDs (required, comma-separated):

2
docs/src/components/DockerComposeGenerator/generator/index.ts
View File

@ -123,7 +123,7 @@ function buildEnvironment(
function buildDeploy(device: DeviceConfig, input: GeneratorInput): string[] { function buildDeploy(device: DeviceConfig, input: GeneratorInput): string[] {
if (device.id === "stable-tensorrt") { if (device.id === "stable-tensorrt") {
const count = input.nvidiaGpuCount || "all"; const count = input.nvidiaGpuCount || "all";
const isAll = count.toLowerCase() === "all"; const isAll = count === "all";
const deviceId = input.nvidiaGpuDeviceId?.trim(); const deviceId = input.nvidiaGpuDeviceId?.trim();
if (isAll) { if (isAll) {

20
docs/src/components/DockerComposeGenerator/hooks/useConfigGenerator.ts
View File

@ -29,7 +29,7 @@ export function useConfigGenerator() {
return initial; return initial;
}); });
const [nvidiaGpuCount, setNvidiaGpuCount] = useState("all"); const [nvidiaGpuCount, setNvidiaGpuCount] = useState("");
const [nvidiaGpuDeviceId, setNvidiaGpuDeviceId] = useState(""); const [nvidiaGpuDeviceId, setNvidiaGpuDeviceId] = useState("");
const [configPath, setConfigPath] = useState(""); const [configPath, setConfigPath] = useState("");
const [mediaPath, setMediaPath] = useState(""); const [mediaPath, setMediaPath] = useState("");
@ -56,7 +56,7 @@ export function useConfigGenerator() {
} }
return next; return next;
}); });
setNvidiaGpuCount("all"); setNvidiaGpuCount("");
setNvidiaGpuDeviceId(""); setNvidiaGpuDeviceId("");
setGpuDeviceIdError(false); setGpuDeviceIdError(false);
}, []); }, []);
@ -121,15 +121,13 @@ export function useConfigGenerator() {
); );
const handleNvidiaGpuCountChange = useCallback((value: string) => { const handleNvidiaGpuCountChange = useCallback((value: string) => {
const lower = value.trim().toLowerCase(); // Only allow digits
if (lower === "all" || lower === "" || /^[0-9]+$/.test(lower)) { setNvidiaGpuCount(value);
setNvidiaGpuCount(lower || "all"); if (value === "") {
if (lower === "all") { setNvidiaGpuDeviceId("");
setNvidiaGpuDeviceId(""); setGpuDeviceIdError(false);
setGpuDeviceIdError(false); } else {
} else if (/^[0-9]+$/.test(lower)) { setGpuDeviceIdError(false);
setGpuDeviceIdError(false);
}
} }
}, []); }, []);