mirror of
https://gitlab.com/nonguix/nonguix.git
synced 2024-11-25 09:58:09 +01:00
8a56ebb666
* nongnu/services/nvidia.scm (nvidia-service-type): Don't extend kernel-module-loader-service-type, move this part to… (nvidia-shepherd-service)[start]: …here. Check driver and command availablity. [stop]: Likewise. [modules]: Add (guix build utils) for ‘invoke/quiet’. Add (rnrs io ports) for ‘get-line’. Signed-off-by: John Kehayias <john.kehayias@protonmail.com>
64 lines
2.7 KiB
Scheme
64 lines
2.7 KiB
Scheme
;;; SPDX-License-Identifier: GPL-3.0-or-later
|
|
;;; Copyright © 2022, 2024 Hilton Chain <hako@ultrarare.space>
|
|
|
|
(define-module (nongnu services nvidia)
|
|
#:use-module (guix gexp)
|
|
#:use-module (guix records)
|
|
#:use-module (gnu packages)
|
|
#:use-module (gnu packages linux)
|
|
#:use-module (gnu services)
|
|
#:use-module (gnu services base)
|
|
#:use-module (gnu services linux)
|
|
#:use-module (gnu services shepherd)
|
|
#:use-module (nongnu packages nvidia)
|
|
#:export (nvidia-configuration
|
|
nvidia-configuration?
|
|
nvidia-configuration-record?
|
|
nvidia-service-type))
|
|
|
|
(define-record-type* <nvidia-configuration>
|
|
nvidia-configuration make-nvidia-configuration
|
|
nvidia-configuration?
|
|
(driver nvidia-configuration-driver
|
|
(default nvda)) ; file-like
|
|
(firmware nvidia-configuration-firmware
|
|
(default nvidia-firmware)) ; file-like
|
|
(module nvidia-configuration-module
|
|
(default nvidia-module))) ; file-like
|
|
|
|
(define (nvidia-shepherd-service config)
|
|
(let ((nvidia-driver (nvidia-configuration-driver config))
|
|
(nvidia-smi (file-append nvidia-driver "/bin/nvidia-smi")))
|
|
(list (shepherd-service
|
|
(documentation "Prepare system environment for NVIDIA driver.")
|
|
(provision '(nvidia))
|
|
(requirement '(udev))
|
|
(one-shot? #t)
|
|
(modules '(((guix build utils) #:select (invoke/quiet))
|
|
((rnrs io ports) #:select (get-line))))
|
|
(start
|
|
#~(lambda _
|
|
(when (file-exists? "/proc/driver/nvidia")
|
|
(let ((modprobe (call-with-input-file
|
|
"/proc/sys/kernel/modprobe" get-line)))
|
|
(false-if-exception
|
|
(begin
|
|
(invoke/quiet modprobe "--" "nvidia_uvm")
|
|
(invoke/quiet #$nvidia-smi)))))))))))
|
|
|
|
(define nvidia-service-type
|
|
(service-type
|
|
(name 'nvidia)
|
|
(extensions
|
|
(list (service-extension shepherd-root-service-type
|
|
nvidia-shepherd-service)
|
|
(service-extension profile-service-type
|
|
(compose list nvidia-configuration-driver))
|
|
(service-extension udev-service-type
|
|
(compose list nvidia-configuration-driver))
|
|
(service-extension firmware-service-type
|
|
(compose list nvidia-configuration-firmware))
|
|
(service-extension linux-loadable-module-service-type
|
|
(compose list nvidia-configuration-module))))
|
|
(default-value (nvidia-configuration))
|
|
(description "Prepare system environment for NVIDIA driver.")))
|