numbus/gpu-detect
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

merge into: numbus/gpu-detect:unstable
Branches Tags
numbus/gpu-detect:26.05 numbus/gpu-detect:ids-database numbus/gpu-detect:unstable
..
pull from: numbus/gpu-detect:26.05
Branches Tags
numbus/gpu-detect:ids-database numbus/gpu-detect:unstable numbus/gpu-detect:26.05
1 Commits
unstable .. 26.05
Author SHA1 Message Date
Raphaël Numbus cafe938569 Changes to the script. Removed the IDs, now in a dedicated ids-database branch. 2026-07-24 12:00:52 +02:00
3 changed files with 166 additions and 208 deletions
Expand all files Collapse all files
client/gpu-detect.sh
Executable → Regular
+132 -174
View File
@@ -9,100 +9,98 @@
# it under the terms of the GNU General Public License as published by # it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or # the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version. # (at your option) any later version.
#
# This package is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <https://www.gnu.org/licenses/>
# Avoid cryptic failure when running dash on this script # Avoid cryptic failure when running dash on this script
shopt -s compat31 nocasematch 2>/dev/null || { echo "Error: this script only works with bash." && exit 1; } shopt -s compat31 nocasematch 2>/dev/null || { echo "Error: this script only works with bash." && exit; }
# Colors
GREEN='\033[0;32m'
BLUE='\033[0;34m'
YELLOW='\033[1;33m'
RED='\033[0;31m'
NC='\033[0m' # No Color
# Defining variables
USER_ID="$(id -u)" USER_ID="$(id -u)"
PCI_IDS_LOCATION="/run/user/${USER_ID}/gpu-detect" PCI_IDS_LOCATION="/run/user/${USER_ID}/gpu-detect"
declare -A args=( declare -A args=(
["brand"]="all" ["brand"]="all"
["hw_accel"]="true" ["hw_accel"]="true"
) )
PCI_IDS=()
show_help() {
gum style --border-foreground="030" --bold --border=normal --padding="0 2" "GPU detection tool for NixOS" gum style --border-foreground="030" --bold --border=normal --padding="0 2" "GPU detection tool for NixOS"
echo "Usage: gpu-detect [OPTIONS] [PCIID]..."
echo ""
echo "Options:"
echo " -b, --brand BRAND Select GPU brand to detect: all, amd, intel, nvidia (default: all)"
echo " -hw, --hw_accel BOOL Provide hardware acceleration configuration: true, false (default: true)"
echo " -h, --help Show this help message and exit"
echo ""
echo "For each GPU installed in your system (or matching the specified brand/PCI ID),"
echo "the program will provide you the needed NixOS configuration."
exit 0
}
while [[ $# -gt 0 ]]; do while [[ $# -gt 0 ]]; do
case "${1}" in case "${1}" in
--brand|-b) --brand|-b)
if [[ -n "${2}" && ! "${2}" =~ ^- ]]; then args["brand"]="${2}"
args["brand"]="${2,,}"
shift 2 shift 2
else
echo -e "${RED}Error: --brand requires an argument.${NC}"
exit 1
fi
;; ;;
--hw_accel|-hw) --hw_accel|-hw)
if [[ -n "${2}" && ! "${2}" =~ ^- ]]; then args["hw_accel"]="${2}"
args["hw_accel"]="${2,,}"
shift 2 shift 2
else
echo -e "${RED}Error: --hw_accel requires an argument.${NC}"
exit 1
fi
;; ;;
--help|-h) --help|-h|*)
show_help echo "Usage: gpu-detect --brand=[all/amd/intel/nvidia] --hw_accel=[true/false] [PCIID]..."
;; echo ""
-*) echo "For each GPU installed in your system, the program"
echo -e "${RED}Unknown option ${1}${NC}" echo "will provide you the needed NixOS configuration."
show_help exit 0
;;
*)
PCI_IDS+=("${1}")
shift
;; ;;
esac esac
done done
gum style --border-foreground="030" --bold --border=normal --padding="0 2" "GPU detection tool for NixOS" gpu_detect() {
# Get the PCI IDs from the repo
# Get the PCI IDs from the repo if [[ ! -d "${PCI_IDS_LOCATION}" ]]; then
if [[ ! -d "${PCI_IDS_LOCATION}" ]]; then
echo -e "${BLUE}Fetching latest PCI IDs database...${NC}"
mkdir -p "${PCI_IDS_LOCATION}" mkdir -p "${PCI_IDS_LOCATION}"
if ! git clone -q -b ids-database https://gittea.dev/numbus/gpu-detect.git "${PCI_IDS_LOCATION}"; then git clone -q -b ids-database https://gittea.dev/numbus/gpu-detect.git "${PCI_IDS_LOCATION}"
# Fallback to local checkout if remote is not available for testing purposes else
if [[ -d "../.git" ]]; then if ! git -C "${PCI_IDS_LOCATION}" pull; then
git clone -q -b ids-database "file://$(realpath ../)" "${PCI_IDS_LOCATION}" || true echo "Directory ${PCI_IDS_LOCATION} is not empty and does not"
fi echo "contain the repository. Please check the folder clean up manually."
fi
else
echo -e "${BLUE}Updating PCI IDs database...${NC}"
if ! git -C "${PCI_IDS_LOCATION}" pull -q; then
echo -e "${RED}Error: Directory ${PCI_IDS_LOCATION} is not empty and does not contain the repository. Please check the folder and clean up manually.${NC}"
exit 1 exit 1
fi fi
fi fi
nvidia_detect() { amd_detect() {
local PCI_ID="${1}"
local NVIDIA_VERSION=""
}
intel_detect() {
}
nvidia_detect() {
PCI_ID=${1}
NVIDIA_VERSION=""
local NVIDIA_TEXT="""
{ config, lib, pkgs, ... }:
{
hardware.graphics = {
enable = true;
extraPackages = with pkgs; [ ${EXTRA_PACKAGES} ];
};
services.xserver.videoDrivers = [ "nvidia" ];
hardware.nvidia = {
modesetting.enable = true;
nvidiaSettings = true;
package = config.boot.kernelPackages.nvidiaPackages.${NVIDIA_VERSION};
# Unsupported options
powerManagement.enable = false;
powerManagement.finegrained = false;
open = false;
};
}
"""
local VERSIONS=("unsupported" "legacy_340" "legacy_390" "legacy_470" "legacy_580" "current") local VERSIONS=("unsupported" "legacy_340" "legacy_390" "legacy_470" "legacy_580" "current")
for version in "${VERSIONS[@]}"; do
for version in ${VERSIONS[@]}; do
if grep -q -i "${PCI_ID}" "${PCI_IDS_LOCATION}/nvidia/${version}.ids" 2>/dev/null; then if grep -q -i "${PCI_ID}" "${PCI_IDS_LOCATION}/nvidia/${version}.ids" 2>/dev/null; then
NVIDIA_VERSION="${version}" NVIDIA_VERSION="${version}"
break break
@@ -110,145 +108,105 @@ nvidia_detect() {
done done
if [[ -z "${NVIDIA_VERSION}" ]]; then if [[ -z "${NVIDIA_VERSION}" ]]; then
gum style --foreground="212" "Uh oh. It seems that your card couldn't be detected." "A newer driver may add support for your card later." echo "Uh oh. It seems that your card couldn't be detected."
echo "A newer driver may add support for your card later."
return 1 return 1
fi fi
if [[ "${NVIDIA_VERSION}" == "unsupported" ]]; then if [[ "${NVIDIA_VERSION}" == "unsupported" ]]; then
gum style --foreground="212" "Good news! You have nothing to do, your card runs with the 'nouveau' driver that is integrated in the kernel." "" "It seems you have a very old card! The proprietary NVIDIA drivers have been removed from Nixpkgs therefore it is not possible to install them." echo "Good news! You have nothing to do, your card runs with the"
return 0 echo "\"nouveau\" driver that is integrated in the kernel."
fi echo ""
echo "It seems you have a very old card!"
local EXTRA_PACKAGES="" echo "The proprietary NVIDIA drivers have been removed from Nixpkgs"
local HW_ACCEL_CONFIG="" echo "therefore it is not possible to install them."
if [[ "${args["hw_accel"]}" == "true" ]]; then elif [[ "${NVIDIA_VERSION}" == "legacy_340" || "${NVIDIA_VERSION}" == "legacy_390" || "${NVIDIA_VERSION}" == "legacy_470" ]]; then
if [[ "${NVIDIA_VERSION}" == "legacy_340" || "${NVIDIA_VERSION}" == "legacy_390" || "${NVIDIA_VERSION}" == "legacy_470" ]]; then echo "An official legacy NVIDIA driver release is available for"
EXTRA_PACKAGES="libvdpau" echo "your card."
elif [[ "${NVIDIA_VERSION}" == "legacy_580" || "${NVIDIA_VERSION}" == "current" ]]; then echo ""
EXTRA_PACKAGES="nvidia-vaapi-driver libvdpau" echo "You can add the following code snippet to your NixOS configuration"
fi echo "in order to enable the proprietary NVIDIA driver with VDPAU support."
HW_ACCEL_CONFIG=" EXTRA_PACKAGES="\"libvdpau\""
echo ${NVIDIA_TEXT}
elif [[ "${NVIDIA_VERSION}" == "legacy_580" ]]; then
echo "An official legacy NVIDIA driver release is available for"
echo "your card."
echo ""
echo "You can add the following nix code to your NixOS configuration"
echo "in order to enable the proprietary NVIDIA driver with VDPAU and"
echo "VAAPI (if applicable) support."
EXTRA_PACKAGES="\"nvidia-vaapi-driver\" \"libvdpau\""
echo ${NVIDIA_TEXT}
elif [[ "${NVIDIA_VERSION}" == "current" ]]; then
echo "An official stable NVIDIA driver release is available for"
echo "your card."
echo ""
echo "You can add the following nix code to your NixOS configuration"
echo "in order to enable the proprietary NVIDIA driver with hardware"
echo "acceleration support."
echo """
{ config, lib, pkgs, ... }:
{
hardware.graphics = { hardware.graphics = {
enable = true; enable = true;
extraPackages = with pkgs; [ ${EXTRA_PACKAGES} ]; extraPackages = with pkgs; [ ${EXTRA_PACKAGES} ];
};" };
fi services.xserver.videoDrivers = [ "nvidia" ];
local DRIVER_INFO=""
local NVIDIA_CONFIG=""
local NIX_PACKAGE_VER="${NVIDIA_VERSION}"
if [[ "${NVIDIA_VERSION}" == "current" ]]; then
NIX_PACKAGE_VER="stable"
fi
if [[ "${NVIDIA_VERSION}" == "current" ]]; then
DRIVER_INFO="An official stable NVIDIA driver release is available for your card."
NVIDIA_CONFIG="{ config, lib, pkgs, ... }:
{${HW_ACCEL_CONFIG}
services.xserver.videoDrivers = [ \"nvidia\" ];
hardware.nvidia = { hardware.nvidia = {
modesetting.enable = true; modesetting.enable = true;
nvidiaSettings = true; nvidiaSettings = true;
open = true; open = true;
package = config.boot.kernelPackages.nvidiaPackages.${NIX_PACKAGE_VER}; package = config.boot.kernelPackages.nvidiaPackages.${NVIDIA_VERSION};
# Advanced options, tweak if needed # Advanced options, tweak if needed
powerManagement.enable = false; # Experimental: Enable this if you have graphical corruption issues or application crashes after waking up from sleep. powerManagement.enable = false; # Experimental: Enable this if you have graphical corruption issues or application crashes after waking up from sleep.
powerManagement.finegrained = false; # Experimental: Turns off GPU when not in use powerManagement.finegrained = false; # Experimental: Turns off GPU when not in use
}; };
}" }
else """
DRIVER_INFO="An official legacy NVIDIA driver release is available for your card." fi
NVIDIA_CONFIG="{ config, lib, pkgs, ... }: }
{${HW_ACCEL_CONFIG}
services.xserver.videoDrivers = [ \"nvidia\" ];
hardware.nvidia = {
modesetting.enable = true;
nvidiaSettings = true;
package = config.boot.kernelPackages.nvidiaPackages.${NIX_PACKAGE_VER};
# Unsupported options
powerManagement.enable = false; if [ -z "${1}" ]; then
powerManagement.finegrained = false;
open = false;
}; NV_DEVICES=$(lspci -mn | awk '{ gsub("\"",""); if ((${2} ~ "030[0-2]") && ($3 == "10de" || $3 == "12d2")) { print ${1} } }')
}"
if [ -z "$NV_DEVICES" ]; then
echo "No NVIDIA GPU detected."
exit 0
fi fi
gum style --foreground="212" "${DRIVER_INFO}" echo "Detected NVIDIA GPUs:"
if [[ "${args["hw_accel"]}" == "true" ]]; then
gum style "You can add the following code snippet to your NixOS configuration in order to enable the proprietary NVIDIA driver with hardware acceleration support."
else
gum style "You can add the following code snippet to your NixOS configuration in order to enable the proprietary NVIDIA driver."
fi
gum format '```nix
'"${NVIDIA_CONFIG}"'
```'
}
# Main logic
if [[ ${#PCI_IDS[@]} -gt 0 ]]; then
for id in "${PCI_IDS[@]}" ; do
# Try to parse NVIDIA PCI ID (vendor 10de)
PCIID=$(echo "$id" | sed -rn 's/^(10de)?:?([0-9a-fA-F]{4})$/10de\2/ip')
if [ -n "$PCIID" ]; then
gum style --foreground="030" "Checking driver support for NVIDIA PCI ID [$(echo $PCIID | sed -r 's/(....)(....)/\1:\2/')]"
nvidia_detect "$PCIID"
else
echo -e "${YELLOW}Currently only NVIDIA PCI IDs (10de:XXXX) are supported when passing explicitly.${NC}"
fi
done
else
if [[ "${args["brand"]}" == "all" || "${args["brand"]}" == "nvidia" ]]; then
NV_DEVICES=$(lspci -mn 2>/dev/null | awk '{ gsub("\"",""); if (($2 ~ "030[0-2]") && ($3 == "10de" || $3 == "12d2")) { print $1 } }')
if [ -n "$NV_DEVICES" ]; then
gum style --foreground="030" "Detected NVIDIA GPUs:"
for d in $NV_DEVICES ; do for d in $NV_DEVICES ; do
lspci -nn -s $d lspci -nn -s $d
done done
for d in $NV_DEVICES ; do for d in $NV_DEVICES ; do
gum style --foreground="030" "Checking card: $(lspci -s $d | awk -F: '{print $3}')" echo -e "\nChecking card: $(lspci -s $d | awk -F: '{print $3}')"
nvidia_detect "$(lspci -mn -s "$d" | awk '{ gsub("\"",""); print $3 $4 }')" nvidia_detect "$(lspci -mn -s "$d" | awk '{ gsub("\"",""); print $3 $4 }')"
done done
else else
if [[ "${args["brand"]}" == "nvidia" ]]; then
gum style --foreground="212" "No NVIDIA GPU detected." for id in "$@" ; do
fi PCIID=$(echo "$id" | sed -rn 's/^(10de)?:?([0-9a-fA-F]{4})$/10de\2/ip')
fi if [ -z "$PCIID" ]; then
echo "Error parsing PCI ID '$id'."
exit 2
fi fi
if [[ "${args["brand"]}" == "all" || "${args["brand"]}" == "amd" ]]; then echo "Checking driver support for PCI ID [$(echo $PCIID | sed -r 's/(....)(....)/\1:\2/')]"
AMD_DEVICES=$(lspci -mn 2>/dev/null | awk '{ gsub("\"",""); if (($2 ~ "030[0-2]") && ($3 == "1002")) { print $1 } }') nvidia_detect "$PCIID"
if [ -n "$AMD_DEVICES" ]; then
gum style --foreground="030" "Detected AMD GPUs:"
for d in $AMD_DEVICES ; do
lspci -nn -s $d
done done
gum style --foreground="212" "AMD configuration generation is not fully implemented yet."
else
if [[ "${args["brand"]}" == "amd" ]]; then
gum style --foreground="212" "No AMD GPU detected."
fi
fi
fi fi
if [[ "${args["brand"]}" == "all" || "${args["brand"]}" == "intel" ]]; then
INTEL_DEVICES=$(lspci -mn 2>/dev/null | awk '{ gsub("\"",""); if (($2 ~ "030[0-2]") && ($3 == "8086")) { print $1 } }')
if [ -n "$INTEL_DEVICES" ]; then
gum style --foreground="030" "Detected Intel GPUs:"
for d in $INTEL_DEVICES ; do
lspci -nn -s $d
done
gum style --foreground="212" "Intel configuration generation is not fully implemented yet."
else
if [[ "${args["brand"]}" == "intel" ]]; then
gum style --foreground="212" "No Intel GPU detected."
fi
fi
fi
fi
exit 0 }
exit $RET
server/intel.md → intel.md
View File
server/nvidia.md → nvidia.md
View File