Skip to content

Commit

Permalink
Include Ubuntu NVIDIA dependencies to CI.
Browse files Browse the repository at this point in the history
  • Loading branch information
0mdc committed Nov 18, 2024
1 parent 4e0e20d commit 79545f2
Show file tree
Hide file tree
Showing 2 changed files with 17 additions and 0 deletions.
16 changes: 16 additions & 0 deletions .github/actions/install_ubuntu_gpu_deps/action.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
# This action installs required Ubuntu NVIDIA dependencies.
# It assumes that `nvidia-smi` is available on the system.
name: "Install GPU Dependencies"
runs:
using: composite
steps:
- name: Install GPU Dependencies
shell: bash
run: |
NVIDIA_SMI_OUTPUT=$(nvidia-smi)
FULL_DRIVER_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K[\d.]+')
MAJOR_DRIVER_VERSION=$(echo "$NVIDIA_SMI_OUTPUT" | grep -oP 'Driver Version:\s+\K\d+')
UBUNTU_PACKAGE_SUFFIX="${MAJOR_DRIVER_VERSION}=${FULL_DRIVER_VERSION}-0ubuntu1"
sudo apt-get update
sudo apt-get install -y libnvidia-common-${UBUNTU_PACKAGE_SUFFIX}
sudo apt-get install -y libnvidia-gl-${UBUNTU_PACKAGE_SUFFIX}
1 change: 1 addition & 0 deletions .github/workflows/install_and_test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -64,6 +64,7 @@ jobs:
- name: CPU info
run: cat /proc/cpuinfo
- uses: "./habitat-sim/.github/actions/install_ubuntu_deps"
- uses: "./habitat-sim/.github/actions/install_ubuntu_gpu_deps"
- name: Build, install habitat-sim
run: |-
#give cmake ownership to the runner for installation
Expand Down

0 comments on commit 79545f2

Please sign in to comment.