zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use the correct repo for arm64 CUDA

Browse Source
This commit is contained in:
Cheng
2025-11-18 09:34:12 +09:00
parent 82faffa7bb
commit 0f8c8c8a11
2 changed files with 11 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
.github/actions/setup-linux/action.yml vendored
View File

@@ -60,17 +60,13 @@ runs:
"cuda-13.0": "libcudnn9-dev-cuda-13 cuda-toolkit-13-0"
}
run: |
export ARCH=${{ runner.arch == 'arm64' && 'arm64' || 'x86_64' }}
export ARCH=${{ runner.arch == 'arm64' && 'sbsa' || 'x86_64' }}
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/$ARCH/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 ${{ fromJson(env.PACKAGES)[inputs.toolkit] }}
- name: Install NCCL
# For some reason NVIDIA did not provide arm64 package for libnccl2.
if: ${{ startsWith(inputs.toolkit, 'cuda') && runner.arch != 'arm64' }}
shell: bash
run: sudo apt-get install -y libnccl2 libnccl-dev
sudo apt-get install -y \
libnccl2 libnccl-dev \
${{ fromJson(env.PACKAGES)[inputs.toolkit] }}
- name: CUDA packages and driver report
if: ${{ startsWith(inputs.toolkit, 'cuda') }}