Ubuntu-CloudInit-Docs/samples/ubuntu/ubuntu-resolute-cloudinit+nvidia+runtime.sh at main · UntouchedWagons/Ubuntu-CloudInit-Docs · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
#! /bin/bash

set -xe

VMID="${VMID:-8202}"
STORAGE="${STORAGE:-local-zfs}"

IMG="resolute-server-cloudimg-amd64.img"
BASE_URL="https://cloud-images.ubuntu.com/resolute/current"
EXPECTED_SHA=$(wget -qO- "$BASE_URL/SHA256SUMS" | awk '/'$IMG'/{print $1}')

download() {
    wget -q "$BASE_URL/$IMG"
}

verify() {
    sha256sum "$IMG" | awk '{print $1}'
}

[ ! -f "$IMG" ] && download

ACTUAL_SHA=$(verify)

if [ "$EXPECTED_SHA" != "$ACTUAL_SHA" ]; then
    rm -f "$IMG"
    download
    ACTUAL_SHA=$(verify)
    [ "$EXPECTED_SHA" != "$ACTUAL_SHA" ] && exit 1
fi

rm -f resolute-server-cloudimg-amd64-resized.img
cp resolute-server-cloudimg-amd64.img resolute-server-cloudimg-amd64-resized.img
qemu-img resize resolute-server-cloudimg-amd64-resized.img 8G

sudo qm destroy $VMID || true
sudo qm create $VMID --name "ubuntu-resolute-template-nvidia-runtime" --ostype l26 \
    --memory 1024 --balloon 0 \
    --agent 1 \
    --bios ovmf --machine q35 --efidisk0 $STORAGE:0,pre-enrolled-keys=0 \
    --cpu host --socket 1 --cores 1 \
    --vga serial0 --serial0 socket  \
    --net0 virtio,bridge=vmbr0
sudo qm importdisk $VMID resolute-server-cloudimg-amd64-resized.img $STORAGE
sudo qm set $VMID --scsihw virtio-scsi-pci --virtio0 $STORAGE:vm-$VMID-disk-1,discard=on
sudo qm set $VMID --boot order=virtio0
sudo qm set $VMID --scsi1 $STORAGE:cloudinit

if [ ! -d "/var/lib/vz/snippets" ]; then
  mkdir -p "/var/lib/vz/snippets"
fi

cat << EOF | sudo tee /var/lib/vz/snippets/ubuntu-resolute-runtime.yaml
#cloud-config
runcmd:
    - curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg
    - curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
    - apt-get update
    # NOTE: Verify nvidia-dkms-590-server and nvidia-utils-590-server are available for Ubuntu 26.04.
    # Check available versions with: apt-cache search nvidia-dkms
    - apt-get install -y qemu-guest-agent nvidia-dkms-590-server nvidia-utils-590-server nvidia-container-runtime
    - systemctl enable ssh
    - reboot
# Taken from https://forum.proxmox.com/threads/combining-custom-cloud-init-with-auto-generated.59008/page-3#post-428772
EOF

echo "timezone: "$(cat /etc/timezone) | sudo tee -a /var/lib/vz/snippets/ubuntu-resolute-runtime.yaml
echo "locale: "$LANG | sudo tee -a /var/lib/vz/snippets/ubuntu-resolute-runtime.yaml

sudo qm set $VMID --cicustom "vendor=local:snippets/ubuntu-resolute-runtime.yaml"
sudo qm set $VMID --tags ubuntu-template,resolute,cloudinit,nvidia
sudo qm set $VMID --ciuser $USER
sudo qm set $VMID --sshkeys ~/.ssh/authorized_keys
sudo qm set $VMID --ipconfig0 ip=dhcp,ip6=dhcp
sudo qm template $VMID