We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
nvcc
CMAKE_ARGS
1 parent e3b4475 commit 59c30baCopy full SHA for 59c30ba
1 file changed
vllm-tensorizer/Dockerfile
@@ -130,6 +130,7 @@ RUN --mount=type=bind,from=vllm-downloader,source=/git/vllm,target=/workspace,rw
130
--mount=type=tmpfs,target=/tmp \
131
. /opt/sccache-start.sh && \
132
. /opt/arch_flags.sh && \
133
+ export CMAKE_ARGS='-DCMAKE_CUDA_COMPILER=/opt/nvcc-wrapper.py' && \
134
if [ -z "$MAX_JOBS" ]; then unset MAX_JOBS; fi && \
135
python3 -m pip install --no-cache-dir py-cpuinfo 'cmake>=3.26.1,<4' grpcio-tools && \
136
if [ -f 'use_existing_torch.py' ]; then \
0 commit comments