mirror of
https://github.com/coollabsio/coolify.git
synced 2026-11-06 13:10:37 -04:00
48 lines
2.3 KiB
PHP
48 lines
2.3 KiB
PHP
<?php
|
|
|
|
namespace App\Actions\Development;
|
|
|
|
use App\Models\Server;
|
|
use Lorisleiva\Actions\Concerns\AsAction;
|
|
|
|
/**
|
|
* Installs the NVIDIA driver and the NVIDIA Container Toolkit on an Ubuntu development VM with a passed-through GPU,
|
|
* then refreshes the server details so Coolify shows the GPU. Steps that are already done are skipped.
|
|
*
|
|
* @see https://ubuntu.com/server/docs/how-to/graphics/install-nvidia-drivers/
|
|
* @see https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
|
|
*/
|
|
class InstallDevelopmentGpuDriver
|
|
{
|
|
use AsAction;
|
|
|
|
public function handle(Server $server): void
|
|
{
|
|
instant_remote_process([
|
|
'set -e',
|
|
'export DEBIAN_FRONTEND=noninteractive',
|
|
'if ! nvidia-smi >/dev/null 2>&1; then',
|
|
'apt-get update -qq',
|
|
'apt-get install -y -qq ubuntu-drivers-common',
|
|
'ubuntu-drivers install --gpgpu',
|
|
// For example 580-server-open. The utilities package has no -open suffix.
|
|
'flavor=$(dpkg-query -W -f=\'${Package}\n\' \'nvidia-headless-no-dkms-*\' | sed -n \'s/^nvidia-headless-no-dkms-//p\' | head -n1)',
|
|
// ubuntu-drivers installs the modules for the newest kernel, which can differ from the running one.
|
|
'apt-get install -y -qq "nvidia-utils-${flavor%-open}" "linux-modules-nvidia-${flavor}-$(uname -r)"',
|
|
'modprobe nvidia',
|
|
'nvidia-smi',
|
|
'fi',
|
|
'if ! command -v nvidia-ctk >/dev/null 2>&1; then',
|
|
'curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --batch --yes --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg',
|
|
'curl -fsSL https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed \'s#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g\' > /etc/apt/sources.list.d/nvidia-container-toolkit.list',
|
|
'apt-get update -qq',
|
|
'apt-get install -y -qq nvidia-container-toolkit',
|
|
'nvidia-ctk runtime configure --runtime=docker',
|
|
'systemctl restart docker',
|
|
'fi',
|
|
], $server, timeout: 1800, disableMultiplexing: true, maxAttempts: 1);
|
|
|
|
$server->gatherServerMetadata();
|
|
}
|
|
}
|