-
Notifications
You must be signed in to change notification settings - Fork 0
59 lines (55 loc) · 1.74 KB
/
linux_gpu2.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
name: linux-gpu2
on:
workflow_dispatch:
pull_request:
types:
- opened
push:
paths-ignore:
- '.github/workflows/**'
- '!.github/workflows/linux_gpu2.yaml'
- 'README.md'
jobs:
build:
runs-on:
- runs-on
- runner=gpu-nvidia-raw
- run-id=${{ github.run_id }}
env:
CACHE_PATH:
CACHE_KEY:
strategy:
matrix:
arch: [x86_64]
mode: [release]
steps:
- name: Checkout repository
uses: actions/checkout@v4
# - name: Install CUDA Toolkit
# uses: Jimver/cuda-toolkit@v0.2.16
# id: cuda-toolkit
# with:
# cuda: '12.4.0'
# - name: Install CUDA Driver + Toolkit
# run: |
# wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
# sudo dpkg -i cuda-keyring_1.1-1_all.deb
# sudo apt-get update
# sudo apt-get -y install nvidia-open
# sudo modprobe nvidia
# sudo modprobe nvidia-drm
# sudo modprobe nvidia-uvm
# sudo apt-get -y install cuda-toolkit-12-4
# nvidia-smi
- name: Install CUDA Driver
run: |
lspci | grep -i nvidia
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get install -y linux-headers-$(uname -r) build-essential
wget https://us.download.nvidia.com/tesla/550.90.07/NVIDIA-Linux-x86_64-550.90.07.run
chmod +x NVIDIA-Linux-x86_64-550.90.07.run
sudo ./NVIDIA-Linux-x86_64-550.90.07.run --silent
lsmod | grep nvidia
nvidia-smi