1
GBM_BACKEND=nvidia-drm
2
__GLX_VENDOR_LIBRARY_NAME=nvidia
3
ENABLE_VKBASALT=1
4
LIBVA_DRIVER_NAME=nvidia

1
blacklist nouveau
2
options nvidia_drm modeset=1 fbdev=1

1
[Trigger]
2
Operation=Install
3
Operation=Upgrade
4
Operation=Remove
5
Type=Package
6
Target=nvidia
7
Target=linux
8
# Adjust line(6) above to match your driver, e.g. Target=nvidia-470xx-dkms
9
# Change line(7) above, if you are not using the regular kernel For example, Target=linux-lts
10

11
[Action]
12
Description=Update Nvidia module in initcpio
13
Depends=mkinitcpio
14
When=PostTransaction
15
NeedsTargets
16
Exec=/bin/sh -c 'while read -r trg; do case $trg in linux) exit 0; esac; done; /usr/bin/mkinitcpio -P'

1
#!/bin/bash
2

3
# 定义要写入 /etc/environment 的变量
4
nvidia_variables=(
5
    "GBM_BACKEND=nvidia-drm"
6
    "__GLX_VENDOR_LIBRARY_NAME=nvidia"
7
    "ENABLE_VKBASALT=1"
8
    "LIBVA_DRIVER_NAME=nvidia"
9
)
10

11
backup_file() {
12
    local file=$1
13
    sudo cp "$file" "$file.bak"
14
}
15

16
update_environment() {
17
    local env_file="/etc/environment"
18
    backup_file "$env_file"
19

20
    for var in "${nvidia_variables[@]}"; do
21
        if grep -q "^${var%%=*}" "$env_file"; then
22
            sudo sed -i "s|^${var%%=*}.*|$var|" "$env_file"
23
        else
24
            echo "$var" | sudo tee -a "$env_file" > /dev/null
25
        fi
26
    done
27

28
    echo "NVIDIA-specific options have been written to $env_file."
29
}
30

31
update_mkinitcpio_conf() {
32
    local mkinitcpio_conf="/etc/mkinitcpio.conf"
33
    backup_file "$mkinitcpio_conf"
34

35
    sudo sed -i '/^MODULES=/ s/(\(.*\))/(\1 nvidia nvidia_modeset nvidia_uvm nvidia_drm)/' "$mkinitcpio_conf"
36
    sudo sed -i '/^HOOKS=/ s/\(\s*kms\s*\)//g' "$mkinitcpio_conf"
37

38
    echo "mkinitcpio.conf has been updated."
39
}
40

41
create_nvidia_conf() {
42
    local nvidia_conf="/etc/modprobe.d/nvidia.conf"
43
    sudo bash -c "cat > $nvidia_conf" <<EOL
44
blacklist nouveau
45
options nvidia_drm modeset=1 fbdev=1
46
EOL
47

48
    echo "$nvidia_conf has been created."
49
}
50

51
regenerate_initramfs() {
52
    sudo mkinitcpio -P
53
    echo "initramfs has been regenerated."
54
}
55

56
create_nvidia_hook() {
57
    local hook_dir="/etc/pacman.d/hooks"
58
    local hook_file="$hook_dir/nvidia.hook"
59

60
    sudo mkdir -p "$hook_dir"
61

62
    sudo bash -c "cat > $hook_file" <<EOL
63
[Trigger]
64
Operation=Install
65
Operation=Upgrade
66
Operation=Remove
67
Type=Package
68
Target=nvidia
69
Target=linux
70
# Adjust line(6) above to match your driver, e.g. Target=nvidia-470xx-dkms
71
# Change line(7) above, if you are not using the regular kernel For example, Target=linux-lts
72

73
[Action]
74
Description=Update Nvidia module in initcpio
75
Depends=mkinitcpio
76
When=PostTransaction
77
NeedsTargets
78
Exec=/bin/sh -c 'while read -r trg; do case \$trg in linux) exit 0; esac; done; /usr/bin/mkinitcpio -P'
79
EOL
80

81
    echo "nvidia.hook has been added to $hook_dir."
82
}
83

84
main() {
85
    update_environment
86
    update_mkinitcpio_conf
87
    create_nvidia_conf
88
    regenerate_initramfs
89
    create_nvidia_hook
90
}
91

92
main