From de21592151a8aebdb46654895f6fe3255b048238 Mon Sep 17 00:00:00 2001 From: Benson Ma Date: Tue, 2 Dec 2025 13:23:27 +0900 Subject: [PATCH] [cuda] Support installation of nvidia-container-toolkit for Ubuntu 22.04 --- .github/actions/setup-nvidia/action.yml | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/.github/actions/setup-nvidia/action.yml b/.github/actions/setup-nvidia/action.yml index 6d7adc8e70..0c930df125 100644 --- a/.github/actions/setup-nvidia/action.yml +++ b/.github/actions/setup-nvidia/action.yml @@ -61,6 +61,18 @@ runs: ) } + install_nvidia_docker2_ubuntu22() { + ( + set -x + # Install nvidia-driver package if not installed + status="$(dpkg-query -W --showformat='${db:Status-Status}' nvidia-docker2 2>&1)" + if [ ! $? = 0 ] || [ ! "$status" = installed ]; then + sudo apt-get install -y nvidia-container-toolkit-1.17.8 + sudo systemctl restart docker + fi + ) + } + pre_install_nvidia_driver_amzn2() { ( # Purge any nvidia driver installed from RHEL repo @@ -245,6 +257,9 @@ runs: ubuntu20.04) install_nvidia_docker2_ubuntu20 ;; + ubuntu22.04) + install_nvidia_docker2_ubuntu22 + ;; *) echo "ERROR: Unknown distribution ${DISTRIBUTION}" exit 1