Dokploy/dokploy
1
0
Fork 0
mirror of https://github.com/Dokploy/dokploy synced 2025-06-26 18:27:59 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #837 from wish-oss/fix/gpu-setup

Browse Source 
fix: update GPU setup command to use sudo and add error handling
This commit is contained in:
Mauricio Siu
2024-12-08 22:09:23 -06:00
committed by GitHub
parent 96fbfa7ef5 49559ebee6
commit 83b3176f6f
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packages/server/src/utils/gpu-setup.ts
View File

@@ -303,7 +303,7 @@ const setupLocalServer = async (daemonConfig: any) => {
await fs.writeFile(configFile, JSON.stringify(daemonConfig, null, 2)); await fs.writeFile(configFile, JSON.stringify(daemonConfig, null, 2));
const setupCommands = [ const setupCommands = [
`pkexec sh -c ' `sudo sh -c '
cp ${configFile} /etc/docker/daemon.json && cp ${configFile} /etc/docker/daemon.json &&
mkdir -p /etc/nvidia-container-runtime && mkdir -p /etc/nvidia-container-runtime &&
sed -i "/swarm-resource/d" /etc/nvidia-container-runtime/config.toml && sed -i "/swarm-resource/d" /etc/nvidia-container-runtime/config.toml &&
@@ -314,7 +314,14 @@ const setupLocalServer = async (daemonConfig: any) => {
`rm ${configFile}`, `rm ${configFile}`,
].join(" && "); ].join(" && ");
await execAsync(setupCommands); try {
await execAsync(setupCommands);
} catch (error) {
console.error("Setup failed:", error);
throw new Error(
"Failed to configure GPU support. Please ensure you have sudo privileges and try again.",
);
}
}; };
const addGpuLabel = async (nodeId: string, serverId?: string) => { const addGpuLabel = async (nodeId: string, serverId?: string) => {