Use fork of zisk that allows to override CUDA_ARCH (#137)

This commit is contained in:
Han
2025-09-17 20:12:17 +08:00
committed by GitHub
parent 1cb4e9238e
commit 7ef4598594
6 changed files with 53 additions and 9 deletions

View File

@@ -45,6 +45,13 @@ RUN [ -n "$CI" ] || \
apt install -y cuda-toolkit-12-9 && \
apt-get clean && rm -rf /var/lib/apt/lists/*)
# Add nvcc to path
ENV PATH=/usr/local/cuda/bin:$PATH
# Default to build for RTX 50 series
ARG CUDA_ARCH=120
ENV CUDA_ARCH=${CUDA_ARCH}
# Copy the ZisK SDK installer script from the workspace context
COPY scripts/sdk_installers/install_zisk_sdk.sh /tmp/install_zisk_sdk.sh
RUN chmod +x /tmp/install_zisk_sdk.sh