Skip to content

Commit 958bed8

Browse files
authored
Update build-cuda.yml
1 parent 5bfbc5b commit 958bed8

File tree

1 file changed

+2
-25
lines changed

1 file changed

+2
-25
lines changed

.github/workflows/build-cuda.yml

Lines changed: 2 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -21,29 +21,6 @@ jobs:
2121
- name: Install CUDA Toolkit 12.0
2222
run: |
2323
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
3625
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 \

0 commit comments

Comments
 (0)