52 lines
2.2 KiB
Bash
52 lines
2.2 KiB
Bash
#!/usr/bin/env bash
|
|
|
|
# Copyright 2025 The Secureblue Authors
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software distributed under the License is
|
|
# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and limitations under the License.
|
|
|
|
set -oue pipefail
|
|
|
|
nvidia_packages_list=('nvidia-container-toolkit' 'nvidia-driver-cuda')
|
|
if [[ "$IMAGE_NAME" != *"securecore"* ]]; then
|
|
nvidia_packages_list+=('libnvidia-fbc' 'libva-nvidia-driver' 'nvidia-driver' 'nvidia-modprobe' 'nvidia-persistenced' 'nvidia-settings')
|
|
fi
|
|
|
|
curl -L https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo \
|
|
-o /etc/yum.repos.d/nvidia-container-toolkit.repo
|
|
sed -i 's/^gpgcheck=0/gpgcheck=1/' /etc/yum.repos.d/nvidia-container-toolkit.repo
|
|
|
|
curl -L https://negativo17.org/repos/fedora-nvidia.repo -o /etc/yum.repos.d/negativo17-fedora-nvidia.repo
|
|
|
|
|
|
sed -i 's/^enabled=0.*/enabled=1/' /etc/yum.repos.d/nvidia-container-toolkit.repo
|
|
sed -i 's/^enabled=0.*/enabled=1\npriority=90/' /etc/yum.repos.d/negativo17-fedora-nvidia.repo
|
|
# required for rpm-ostree to function properly
|
|
# shellcheck disable=SC2068
|
|
rpm-ostree install ${nvidia_packages_list[@]}
|
|
|
|
kmod_version=$(rpm -qa | grep akmod-nvidia | awk -F':' '{print $(NF)}' | awk -F'-' '{print $(NF-1)}')
|
|
negativo_version=$(rpm -qa | grep nvidia-modprobe | awk -F':' '{print $(NF)}' | awk -F'-' '{print $(NF-1)}')
|
|
|
|
echo "kmod_version: ${kmod_version}"
|
|
echo "negativo_version: ${negativo_version}"
|
|
if [[ "$kmod_version" != "$negativo_version" ]]; then
|
|
echo "Version mismatch!"
|
|
exit 1
|
|
fi
|
|
|
|
curl -L https://raw.githubusercontent.com/NVIDIA/dgx-selinux/master/bin/RHEL9/nvidia-container.pp \
|
|
-o nvidia-container.pp
|
|
semodule -i nvidia-container.pp
|
|
|
|
rm -f nvidia-container.pp
|
|
rm -f /etc/yum.repos.d/negativo17-fedora-nvidia.repo
|
|
rm -f /etc/yum.repos.d/nvidia-container-toolkit.repo
|