github-personal/ack-image-builder
1
0
Fork 0
mirror of https://github.com/outbackdingo/ack-image-builder.git synced 2026-01-27 10:18:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #46 from songkang7/fix_gpu_install

Browse Source 
fix gpu
This commit is contained in:
ringtail
2024-01-04 20:06:06 +08:00
committed by GitHub
parent da5d1a5853 6867d72dae
commit 9e9def090a
8 changed files with 105 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -9,6 +9,8 @@ This repository contains resources and configuration scripts for building a cust
* Aliyun Linux 2 (Alibaba Cloud Linux 2)
* Aliyun Linux 3 (Alibaba Cloud Linux 3)
* CentOS 7.6/7.7/7.8/7.9
* Red Hat Enterprise Linux 9
## Setup

2
scripts/ack-optimized-os-1.20.sh
View File

@@ -184,7 +184,7 @@ source_file() {
preset_gpu() {
if [[ "$PRESET_GPU" == "true" ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
source pkg/run/$RELEASE_VERSION/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then

5
scripts/ack-optimized-os-1.22.sh
View File

@@ -184,8 +184,9 @@ source_file() {
preset_gpu() {
if [[ "$PRESET_GPU" == "true" ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then

5
scripts/ack-optimized-os-1.24.sh
View File

@@ -174,8 +174,9 @@ source_file() {
preset_gpu() {
if [[ "$PRESET_GPU" == "true" ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then

5
scripts/ack-optimized-os-1.26.sh
View File

@@ -73,8 +73,9 @@ source_file() {
preset_gpu() {
if [[ "$PRESET_GPU" == "true" ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then

55
scripts/ack-optimized-os-all.sh
View File

@@ -88,36 +88,43 @@ install_pkg() {
preset_gpu() {
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
if [[ "$PRESET_GPU" != "true" ]]; then
return
fi
if [[ "$PRESET_GPU" == "true" ]]; then
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
return
elif [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -eq 20 ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
source pkg/run/$RELEASE_VERSION/lib/$file_name
done
else
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
fi
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
}
trim_os() {

55
scripts/ack-optimized-os-linux3-all.sh
View File

@@ -91,36 +91,43 @@ install_pkg() {
preset_gpu() {
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
if [[ "$PRESET_GPU" != "true" ]]; then
return
fi
if [[ "$PRESET_GPU" == "true" ]]; then
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
return
elif [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -eq 20 ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
source pkg/run/$RELEASE_VERSION/lib/$file_name
done
else
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
fi
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
}
trim_os() {

55
scripts/ack-optimized-os-rhel9-all.sh
View File

@@ -88,36 +88,43 @@ install_pkg() {
preset_gpu() {
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
if [[ "$PRESET_GPU" != "true" ]]; then
return
fi
if [[ "$PRESET_GPU" == "true" ]]; then
if [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -lt 20 ]]; then
return
elif [[ $(echo "${KUBE_VERSION}" | cut -d. -f2) -eq 20 ]]; then
for file_name in $(ls pkg/run/$RELEASE_VERSION/lib | grep -v init.sh); do
source pkg/run/$RELEASE_VERSION/lib/$file_name
source pkg/run/$RELEASE_VERSION/lib/$file_name
done
else
export SRC_DIR=pkg/run/$RELEASE_VERSION
for file_name in $(ls $SRC_DIR/lib | grep -v init.sh | grep -v common.sh | grep -v log.sh); do
source $SRC_DIR/lib/$file_name
done
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
fi
if [[ $NVIDIA_DRIVER_VERSION == "" ]];then
export NVIDIA_DRIVER_VERSION=460.91.03
fi
nvidia::create_dir
# --nvidia-driver-runfile 指定驱动文件路径
nvidia::prepare_driver_package
# --nvidia-container-toolkit-rpms 指定nvidia container toolkit包含的rpm包所在目录
nvidia::prepare_container_runtime_package
# --nvidia-fabricmanager-rpm 指定nvidia fabric manager安装包（rpm格式）路径
nvidia::prepare_driver_package
# --nvidia-device-plugin-yaml 指定nvidia device plugin yaml文件路径
nvidia::deploy_static_pod
if [[ $RUNTIME == "docker" ]];then
export SKIP_CONTAINER_RUNTIME_CONFIG=true
fi
nvidia::gpu::installer::main
}
trim_os() {