#!/bin/bash

# Script for create a simple ai system with LXC and Incus. 

## Install updated repos for debian 12

echo "Install new repos for debian..."

sudo apt -y install gpg wget

sudo bash -c 'echo "deb http://deb.debian.org/debian bookworm-backports main contrib non-free non-free-firmware" > /etc/apt/sources.list.d/backports.list'

sudo curl -fsSL https://pkgs.zabbly.com/key.asc | gpg --show-keys --fingerprint

sudo mkdir -p /etc/apt/keyrings/

sudo curl -fsSL https://pkgs.zabbly.com/key.asc -o /etc/apt/keyrings/zabbly.asc

sudo sh -c 'cat <<EOF > /etc/apt/sources.list.d/zabbly-incus-stable.sources
Enabled: yes
Types: deb
URIs: https://pkgs.zabbly.com/incus/stable
Suites: $(. /etc/os-release && echo ${VERSION_CODENAME})
Components: main
Architectures: $(dpkg --print-architecture)
Signed-By: /etc/apt/keyrings/zabbly.asc

EOF'

sudo apt install -y software-properties-common 

sudo apt-add-repository -y contrib

sudo apt-add-repository -y non-free

sudo apt-add-repository -y non-free-firmware

sudo apt-get update && sudo apt-get upgrade

# Install new kernel.

sudo apt-get -y install linux-image-6.11.10+bpo-amd64 linux-headers-6.11.10+bpo-amd64

# Install cuda nvidia drivers

echo "Install nvidia drivers"

sudo apt -y purge "*nvidia*"

wget https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get -y -f install
sudo apt -y update

sudo bash -c 'echo "deb [signed-by=/usr/share/keyrings/cuda-archive-keyring.gpg] https://developer.download.nvidia.com/compute/cuda/repos/debian12/x86_64/ /" \
    | tee /etc/apt/sources.list.d/cuda-debian12-x86_64.list'
    
sudo apt -y update

sudo apt -y install cuda-drivers

sudo apt -y install curl
sudo curl -fsSL https://ollama.com/install.sh | sh

sudo systemctl stop ollama && sudo systemctl disable ollama

# Installing incus

echo "Installing Incus"

sudo apt -y install qemu-system-x86/bookworm-backports qemu-system-common/bookworm-backports qemu-system-data/bookworm-backports seabios/bookworm-backports

sudo apt -y install incus

sudo usermod -a -G incus,incus-admin $USER

echo "Please, reboot the server..."