Support installations on amd64 and arm64

This commit is contained in:
Nicolas Mowen 2024-08-07 19:39:43 -06:00
parent 3a64402927
commit 1586722f7e
2 changed files with 39 additions and 4 deletions

View File

@ -1,18 +1,18 @@
target wheels { target wheels {
dockerfile = "docker/main/Dockerfile" dockerfile = "docker/main/Dockerfile"
platforms = ["linux/arm64"] platforms = ["linux/arm64","linux/amd64"]
target = "wheels" target = "wheels"
} }
target deps { target deps {
dockerfile = "docker/main/Dockerfile" dockerfile = "docker/main/Dockerfile"
platforms = ["linux/arm64"] platforms = ["linux/arm64","linux/amd64"]
target = "deps" target = "deps"
} }
target rootfs { target rootfs {
dockerfile = "docker/main/Dockerfile" dockerfile = "docker/main/Dockerfile"
platforms = ["linux/arm64"] platforms = ["linux/arm64","linux/amd64"]
target = "rootfs" target = "rootfs"
} }
@ -23,5 +23,5 @@ target h8l {
deps = "target:deps" deps = "target:deps"
rootfs = "target:rootfs" rootfs = "target:rootfs"
} }
platforms = ["linux/arm64"] platforms = ["linux/arm64","linux/amd64"]
} }

View File

@ -0,0 +1,35 @@
#!/bin/bash
# Update package list and install dependencies
sudo apt-get update
sudo apt-get install -y build-essential cmake git wget linux-modules-extra-$(uname -r)
arch=$(uname -m)
if [[ $arch == "x86_64" ]]; then
sudo apt install -y linux-headers-$(uname -r);
else
sudo apt install -y linux-modules-extra-$(uname -r);
fi
# Clone the HailoRT driver repository
git clone --depth 1 --branch v4.17.0 https://github.com/hailo-ai/hailort-drivers.git
# Build and install the HailoRT driver
cd hailort-drivers/linux/pcie
sudo make all
sudo make install
# Load the Hailo PCI driver
sudo modprobe hailo_pci
# Download and install the firmware
cd ../../
./download_firmware.sh
sudo mv hailo8_fw.4.17.0.bin /lib/firmware/hailo/hailo8_fw.bin
# Install udev rules
sudo cp ./linux/pcie/51-hailo-udev.rules /etc/udev/rules.d/
sudo udevadm control --reload-rules && sudo udevadm trigger
echo "HailoRT driver installation complete."