Skip to content

Commit

Permalink
install cuda
Browse files Browse the repository at this point in the history
  • Loading branch information
multiphaseCFD committed Jan 7, 2025
1 parent 178f3f7 commit 8fbe885
Showing 1 changed file with 4 additions and 1 deletion.
5 changes: 4 additions & 1 deletion .github/workflows/wheel_linux_aarch64_cuda.yml
Original file line number Diff line number Diff line change
Expand Up @@ -88,6 +88,8 @@ jobs:
dnf clean all -y
dnf install gcc-toolset-12 dnf-utils -y
source /opt/rh/gcc-toolset-12/enable -y
yum-config-manager --add-repo http://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa/cuda-rhel8.repo -y
dnf -y install cuda-toolkit-${{ matrix.cuda_version }}-0 git
# ensure nvcc is available
CIBW_ENVIRONMENT: |
Expand All @@ -103,14 +105,15 @@ jobs:
CIBW_TEST_COMMAND: |
DEVICENAME=`echo ${{ matrix.pl_backend }} | sed "s/_/./g"`
ls
if ${{ matrix.pl_backend == 'lightning_gpu'}}
then
pl-device-test --device=${DEVICENAME} --skip-ops -x --tb=short --no-flaky-report
fi
CIBW_REPAIR_WHEEL_COMMAND_LINUX: "./bin/auditwheel repair -w {dest_dir} {wheel}"

CIBW_MANYLINUX_AARCH64_IMAGE: manylinux_2_28
CIBW_MANYLINUX_AARCH64_IMAGE: manylinux_2_28_aarch64

CIBW_BUILD_VERBOSITY: 3

Expand Down

0 comments on commit 8fbe885

Please sign in to comment.