#!/bin/bash

# =====================================================
# Debian 12 安装 CUDA 12.6 一键脚本
# 支持 x86_64 架构 + NVIDIA 显卡
# 注意：建议在 TTY 终端运行（Ctrl+Alt+F3），避免图形界面冲突
# =====================================================

set -e  # 出错即停止

echo "? 开始安装 CUDA 12.6 on Debian 12"

# 检查是否是 root
if [[ $EUID -ne 0 ]]; then
   echo "❌ 此脚本必须以 root 权限运行，请使用 sudo" 
   exit 1
fi

# 检查架构
if [[ $(uname -m) != "x86_64" ]]; then
    echo "❌ 此脚本仅支持 x86_64 架构"
    exit 1
fi

# 1. 卸载旧驱动
echo "? 正在卸载旧版 NVIDIA 驱动..."
apt-get --purge remove -y "*nvidia*" || true
apt-get autoremove -y

# 2. 更新系统并安装依赖
echo "? 更新系统并安装编译工具与内核头文件..."
apt-get update
apt-get install -y build-essential dkms "linux-headers-$(uname -r)"

# === 3. 下载 CUDA 12.6 .deb 安装包 ===
CUDA_REPO_DEB="cuda-repo-debian12-12-6-local_12.6.0-560.28.03-1_amd64.deb"
CUDA_REPO_URL="https://developer.download.nvidia.com/compute/cuda/12.6.0/local_installers/$CUDA_REPO_DEB"

echo "⬇️ 正在下载 CUDA 12.6 安装包..."
cd /tmp
wget -O "$CUDA_REPO_DEB" "$CUDA_REPO_URL"

if [ $? -ne 0 ]; then
    echo "❌ 下载失败，请检查网络或链接是否有效"
    echo "? 提示：可手动访问以下地址下载并重试："
    echo "    https://developer.nvidia.com/cuda-12-6-0-download-archive"
    exit 1
fi

# === 4. 安装 CUDA repo 和 keyring ===
echo "? 安装 CUDA 软件源..."
dpkg -i "$CUDA_REPO_DEB"
cp /var/cuda-repo-debian12-12-6-local/cuda-*-keyring.gpg /usr/share/keyrings/

# === 5. 更新 APT 缓存并安装 CUDA Toolkit（包含驱动）===
echo "? 正在安装 CUDA 12.6 Toolkit（含驱动）..."
apt-get update
apt-get install -y cuda-toolkit-12-6

# 安装驱动（传统闭源模块）
apt-get install -y cuda-drivers

# === 6. 添加环境变量到 ~/.bashrc（对当前用户生效）===
USER_HOME=$(getent passwd $SUDO_USER | cut -d: -f6)
BASHRC="$USER_HOME/.bashrc"

echo "" >> "$BASHRC"
echo "# CUDA 12.6 Environment Variables" >> "$BASHRC"
echo 'export PATH=/usr/local/cuda-12.6/bin:$PATH' >> "$BASHRC"
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH' >> "$BASHRC"

# 同时添加到 root 的 .bashrc（可选）
ROOT_BASHRC="/root/.bashrc"
echo "" >> "$ROOT_BASHRC"
echo "# CUDA 12.6 Environment Variables" >> "$ROOT_BASHRC"
echo 'export PATH=/usr/local/cuda-12.6/bin:$PATH' >> "$ROOT_BASHRC"
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH' >> "$ROOT_BASHRC"

echo "✅ 安装完成！请重启系统以使更改生效：sudo reboot"
echo "? 验证命令：nvidia-smi 和 nvcc -V"        报错  Package 'firmware-nvidia-tesla-gsp' is not installed, so not removed
Package 'firmware-nvidia-graphics' is not installed, so not removed
0 upgraded, 0 newly installed, 0 to remove and 283 not upgraded.
Reading package lists... Done
Building dependency tree... Done
Reading state information... Done
0 upgraded, 0 newly installed, 0 to remove and 283 not upgraded.
? 更新系统并安装编译工具与内核头文件...
Hit:1 http://mirrors.aliyun.com/debian bookworm InRelease
Hit:2 http://mirrors.aliyun.com/debian bookworm-updates InRelease
Hit:3 http://mirrors.aliyun.com/debian-security bookworm-security InRelease
Hit:4 http://mirrors.aliyun.com/debian bookworm-backports InRelease
Hit:5 https://deb.debian.org/debian bookworm InRelease
Hit:6 https://deb.debian.org/debian bookworm-updates InRelease
Hit:7 https://deb.debian.org/debian bookworm-backports InRelease
Hit:8 https://mirrors.aliyun.com/docker-ce/linux/debian bookworm InRelease
Hit:10 http://dl.google.com/linux/chrome/deb stable InRelease
Ign:11 https://packages.microsoft.com/repos/code stable InRelease
Hit:12 https://windsurf-stable.codeiumdata.com/wVxQEIWkwPUEAGf3/apt stable InRelease
Hit:9 https://repo.bookworm.elive.elivecd.org bookworm InRelease
Err:13 https://packages.microsoft.com/repos/code stable Release
  Cannot initiate the connection to packages.microsoft.com:443 (2620:1ec:bdf::73). - connect (101: Network is unreachable) Cannot initiate the connection to packages.microsoft.com:443 (2620:1ec:46::73). - connect (101: Network is unreachable) [IP: 67.205.29.196 443]
Reading package lists... Done
E: The repository 'https://packages.microsoft.com/repos/code stable Release' no longer has a Release file.
N: Updating from such a repository can't be done securely, and is therefore disabled by default.
N: See apt-secure(8) manpage for repository creation and user configuration details.
~/s/driver ❯❯❯

tencent :抱歉，我还没有学会如何回答这个问题，有什么别的我可以帮助你吗？...