Skip to content

Commit dc24641

Browse files
committed
Use NVIDIA container
1 parent 77eddbc commit dc24641

File tree

1 file changed

+4
-41
lines changed

1 file changed

+4
-41
lines changed

.github/workflows/cuda.yml

Lines changed: 4 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -23,52 +23,16 @@ jobs:
2323
fail-fast: false
2424

2525
runs-on: gpu-runner-1
26+
container:
27+
image: nvidia/cuda:12.6.3-devel-ubuntu24.04
28+
options: --gpus all
2629

2730
steps:
2831
- uses: actions/checkout@v4
2932

3033
- name: Install Packages
3134
run: |
32-
sudo apt-get install -y wget
33-
34-
- name: Install CUDA
35-
run: |
36-
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
37-
sudo dpkg -i cuda-keyring_1.1-1_all.deb
38-
sudo apt-get update
39-
sudo apt-get install -y cuda-toolkit
40-
echo "PATH=/usr/local/cuda-12.6/bin:${PATH}" >> $GITHUB_ENV
41-
echo "LD_LIBRARY_PATH=/usr/local/cuda-12.6/lib64:${LD_LIBRARY_PATH}" >> $GITHUB_ENV
42-
echo "CUDA_PATH=/usr/local/cuda-12.6" >> $GITHUB_ENV
43-
44-
# See: https://apt.kitware.com/
45-
- name: Install CMake from Kitware apt
46-
run: |
47-
48-
sudo apt-get update
49-
sudo apt-get install -y ca-certificates gpg gnupg wget
50-
51-
sudo apt-key adv --keyserver keyserver.ubuntu.com --recv-keys 1A127079A92F09ED
52-
53-
test -f /usr/share/doc/kitware-archive-keyring/copyright ||
54-
wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | gpg --dearmor - | sudo tee /usr/share/keyrings/kitware-archive-keyring.gpg >/dev/null
55-
56-
echo 'deb [signed-by=/usr/share/keyrings/kitware-archive-keyring.gpg] https://apt.kitware.com/ubuntu/ jammy main' | sudo tee /etc/apt/sources.list.d/kitware.list >/dev/null
57-
sudo apt-get update
58-
59-
test -f /usr/share/doc/kitware-archive-keyring/copyright ||
60-
sudo rm /usr/share/keyrings/kitware-archive-keyring.gpg
61-
62-
sudo apt-get install -y kitware-archive-keyring
63-
64-
echo 'deb [signed-by=/usr/share/keyrings/kitware-archive-keyring.gpg] https://apt.kitware.com/ubuntu/ jammy-rc main' | sudo tee -a /etc/apt/sources.list.d/kitware.list >/dev/null
65-
sudo apt-get update
66-
67-
sudo apt-get update
68-
sudo apt-get install -y cmake
69-
70-
- name: Check CMake
71-
run: cmake --version
35+
sudo apt-get install -y wget cmake git python3 build-essential
7236
7337
- name: Setup Boost
7438
run: |
@@ -91,7 +55,6 @@ jobs:
9155
git submodule update --init tools/boostdep
9256
python3 tools/boostdep/depinst/depinst.py --git_args "--jobs 3" $LIBRARY
9357
94-
9558
- name: Check CUDA Installation
9659
run: |
9760
nvidia-smi

0 commit comments

Comments
 (0)