File tree Expand file tree Collapse file tree 1 file changed +2
-25
lines changed Expand file tree Collapse file tree 1 file changed +2
-25
lines changed Original file line number Diff line number Diff line change 21
21
- name : Install CUDA Toolkit 12.0
22
22
run : |
23
23
sudo apt update
24
- sudo apt install -y wget gnupg2 lsb-release
25
- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
26
- sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
27
- sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/7fa2af80.pub
28
- sudo add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
29
- sudo apt update
30
- sudo apt install -y cuda-toolkit-12-0
31
-
32
- - name : Install build dependencies
33
- run : |
34
- sudo apt install -y g++-12 ninja-build cmake python3-pip
35
- python3 -m pip install --upgrade pip
24
+ sudo apt install -y wget gnupg2 ca-certificates curl software-properties-common
36
25
37
- - name : Build CUDA wheel
38
- run : |
39
- export PATH=/usr/local/cuda/bin:$PATH
40
- export CUDAToolkit_ROOT=/usr/local/cuda
41
- export CXX=/usr/bin/g++-12
42
- CMAKE_ARGS="-DLLAMA_CUDA=on -DCMAKE_CUDA_HOST_COMPILER=/usr/bin/g++-12 -DCMAKE_CUDA_ARCHITECTURES=89" \
43
- pip wheel . --no-deps --wheel-dir dist
44
-
45
- - name : Upload CUDA wheel as artifact
46
- uses : actions/upload-artifact@v4
47
- with :
48
- name : llama_cpp_cuda_wheel
49
- path : dist/*.whl
26
+ curl -fsSL https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/7fa2af80.pub \
You can’t perform that action at this time.
0 commit comments