Skip to content

Commit 87f3db8

Browse files
author
Mark Saroufim
committed
Fix causal-conv1d build: use --no-build-isolation to avoid CUDA mismatch
uv's build isolation installs torch from default PyPI (cu128) instead of the cu130 system torch, causing the CUDA version check to fail. Using --no-build-isolation makes the build use the system torch.
1 parent 5ce7b4a commit 87f3db8

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

docker/helion.Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,8 @@ RUN sudo uv pip install --system helion
5757
RUN sudo uv pip install --system flash-linear-attention
5858

5959
# Causal Conv1d (Dao-AILab reference for benchmarking)
60-
RUN sudo uv pip install --system causal-conv1d
60+
# --no-build-isolation: use system torch (cu130) instead of build env pulling cu128 from PyPI
61+
RUN sudo uv pip install --system --no-build-isolation causal-conv1d
6162

6263
# # tinygrad
6364
# RUN sudo uv pip install --system tinygrad~=0.10

0 commit comments

Comments
 (0)