feat: Added GPU support feature for Remote Server with setup and status checks, including API endpoints and utility functions

apps/dokploy/components/dashboard/settings/servers/setup-server.tsx

@@ -32,6 +32,7 @@ import Link from "next/link";
 import { useState } from "react";
 import { toast } from "sonner";
 import { ShowDeployment } from "../../application/deployments/show-deployment";
+import { GPUSupport } from "./gpu-support";
 
 interface Props {
 	serverId: string;
@@ -89,9 +90,10 @@ export const SetupServer = ({ serverId }: Props) => {
 				) : (
 					<div id="hook-form-add-gitlab" className="grid w-full gap-1">
 						<Tabs defaultValue="ssh-keys">
-							<TabsList className="grid grid-cols-2 w-[400px]">
+							<TabsList className="grid grid-cols-3 w-[400px]">
 								<TabsTrigger value="ssh-keys">SSH Keys</TabsTrigger>
 								<TabsTrigger value="deployments">Deployments</TabsTrigger>
+								<TabsTrigger value="gpu-setup">GPU Setup</TabsTrigger>
 							</TabsList>
 							<TabsContent
 								value="ssh-keys"
@@ -291,6 +293,14 @@ export const SetupServer = ({ serverId }: Props) => {
 									</div>
 								</CardContent>
 							</TabsContent>
+							<TabsContent
+								value="gpu-setup"
+								className="outline-none ring-0 focus-visible:ring-0 focus-visible:ring-offset-0"
+							>
+								<div className="flex flex-col gap-2 text-sm text-muted-foreground pt-3">
+									<GPUSupport serverId={serverId} />
+								</div>
+							</TabsContent>
 						</Tabs>
 					</div>
 				)}

apps/dokploy/server/api/routers/settings.ts

@@ -52,6 +52,10 @@ import {
 	writeMainConfig,
 	writeTraefikConfigInPath,
 } from "@dokploy/server";
+import {
+	checkGPUStatus,
+	setupGPUSupport,
+} from "@dokploy/server/src/utils/gpu-setup";
 import { generateOpenApiDocument } from "@dokploy/trpc-openapi";
 import { TRPCError } from "@trpc/server";
 import { sql } from "drizzle-orm";
@@ -650,6 +654,62 @@ export const settingsRouter = createTRPCRouter({
 		}
 		return { status: "not_cloud" };
 	}),
+	setupGPU: adminProcedure
+		.input(
+			z.object({
+				serverId: z.string(),
+			}),
+		)
+		.mutation(async ({ input }) => {
+			try {
+				if (IS_CLOUD) {
+					return { success: true };
+				}
+
+				if (!input.serverId) {
+					throw new TRPCError({
+						code: "BAD_REQUEST",
+						message: "Server ID is required",
+					});
+				}
+
+				await setupGPUSupport(input.serverId);
+				return { success: true };
+			} catch (error) {
+				throw new TRPCError({
+					code: "INTERNAL_SERVER_ERROR",
+					message:
+						error instanceof Error
+							? error.message
+							: "Failed to enable GPU support",
+					cause: error,
+				});
+			}
+		}),
+	checkGPUStatus: adminProcedure
+		.input(
+			z.object({
+				serverId: z.string().optional(),
+			}),
+		)
+		.query(async ({ input }) => {
+			if (IS_CLOUD) {
+				return {
+					driverInstalled: false,
+					driverVersion: undefined,
+					gpuModel: undefined,
+					runtimeInstalled: false,
+					runtimeConfigured: false,
+					cudaSupport: undefined,
+					cudaVersion: undefined,
+					memoryInfo: undefined,
+					availableGPUs: 0,
+					swarmEnabled: false,
+					gpuResources: 0,
+				};
+			}
+			return await checkGPUStatus(input.serverId);
+		}),
 });
 // {
 // 	"Parallelism": 1,

apps/dokploy/server/api/trpc.ts

@@ -21,8 +21,6 @@ import {
 import type { Session, User } from "lucia";
 import superjson from "superjson";
 import { ZodError } from "zod";
-import { setupGPUSupport } from '@dokploy/server/src/utils/gpu-setup';
-
 /**
  * 1. CONTEXT
  *
@@ -209,10 +207,3 @@ export const adminProcedure = t.procedure.use(({ ctx, next }) => {
 		},
 	});
 });
-
-const appRouter = t.router({
-	setupGPU: t.procedure.mutation(async () => {
-	  await setupGPUSupport();
-	  return { success: true };
-	}),
-  });

apps/dokploy/templates/blender/index.ts

@@ -1,34 +1,34 @@
 import {
-  generateHash,
-  generateRandomDomain,
-  type Template,
-  type Schema,
-  type DomainSchema,
+	type DomainSchema,
+	type Schema,
+	type Template,
+	generateHash,
+	generateRandomDomain,
 } from "../utils";
 
 export function generate(schema: Schema): Template {
-  const mainServiceHash = generateHash(schema.projectName);
-  const mainDomain = generateRandomDomain(schema);
+	const mainServiceHash = generateHash(schema.projectName);
+	const mainDomain = generateRandomDomain(schema);
 
-  const domains: DomainSchema[] = [
-    {
-      host: mainDomain,
-      port: 3000,
-      serviceName: "blender",
-    },
-  ];
+	const domains: DomainSchema[] = [
+		{
+			host: mainDomain,
+			port: 3000,
+			serviceName: "blender",
+		},
+	];
 
-  const envs = [
-    `PUID=1000`,
-    `PGID=1000`,
-    `TZ=Etc/UTC`,
-    `SUBFOLDER=/`,
-    `NVIDIA_VISIBLE_DEVICES=all`,
-    `NVIDIA_DRIVER_CAPABILITIES=all`,
-  ];
+	const envs = [
+		`PUID=1000`,
+		`PGID=1000`,
+		`TZ=Etc/UTC`,
+		`SUBFOLDER=/`,
+		`NVIDIA_VISIBLE_DEVICES=all`,
+		`NVIDIA_DRIVER_CAPABILITIES=all`,
+	];
 
-  return {
-    envs,
-    domains,
-  };
+	return {
+		envs,
+		domains,
+	};
 }

apps/dokploy/templates/templates.ts

@@ -516,7 +516,8 @@ export const templates: TemplateData[] = [
 		id: "blender",
 		name: "Blender",
 		version: "latest",
-		description: "Blender is a free and open-source 3D creation suite. It supports the entire 3D pipeline—modeling, rigging, animation, simulation, rendering, compositing and motion tracking, video editing and 2D animation pipeline.",
+		description:
+			"Blender is a free and open-source 3D creation suite. It supports the entire 3D pipeline—modeling, rigging, animation, simulation, rendering, compositing and motion tracking, video editing and 2D animation pipeline.",
 		logo: "blender.svg",
 		links: {
 			github: "https://github.com/linuxserver/docker-blender",