29 lines
943 B
Bash
29 lines
943 B
Bash
#!/bin/bash
|
|
set -e
|
|
|
|
# Install offline packages required for GPU support
|
|
install_all_offline_packages() {
|
|
echo "Installing GPU driver and toolkit packages"
|
|
# Implementation assumes packages are available locally
|
|
sudo apt-get update
|
|
sudo apt-get install -y nvidia-driver-535 nvidia-headless-535 nvidia-container-toolkit
|
|
}
|
|
|
|
# Deploy the NVIDIA GPU operator
|
|
deploy_plugin() {
|
|
helm repo add nvidia https://helm.ngc.nvidia.com/nvidia || true
|
|
helm upgrade --install gpu-operator nvidia/gpu-operator \
|
|
--namespace gpu-operator \
|
|
--create-namespace \
|
|
--set nodeSelector.kubernetes.io/gpu="true" \
|
|
--set driver.enabled=true \
|
|
--set toolkit.enabled=true \
|
|
--set devicePlugin.enabled=true \
|
|
--set operator.runtimeClass="nvidia-container-runtime" \
|
|
--set operator.defaultRuntime=containerd \
|
|
--set containerRuntime.socketPath=/var/run/containerd/containerd.sock
|
|
}
|
|
|
|
install_all_offline_packages
|
|
deploy_plugin
|