Skip to content

Commit f0ca16f

Browse files
authored
feat: Better cublas support in vLLM container. (#4514)
Signed-off-by: jthomson04 <[email protected]>
1 parent c0e394d commit f0ca16f

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

container/Dockerfile.vllm

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -217,8 +217,8 @@ COPY --from=framework /usr/local/cuda/bin/fatbinary /usr/local/cuda/bin/fatbinar
217217
COPY --from=framework /usr/local/cuda/include/ /usr/local/cuda/include/
218218
COPY --from=framework /usr/local/cuda/nvvm /usr/local/cuda/nvvm
219219
COPY --from=framework /usr/local/cuda/lib64/libcudart.so* /usr/local/cuda/lib64/
220-
COPY --from=framework /usr/local/cuda/lib64/libcublas.so* /usr/local/cuda/lib64/
221-
COPY --from=framework /usr/local/cuda/lib64/libcublasLt.so* /usr/local/cuda/lib64/
220+
RUN ln -s /usr/local/cuda/lib64/libcublas.so.12 /usr/local/cuda/lib64/libcublas.so
221+
RUN ln -s /usr/local/cuda/lib64/libcublasLt.so.12 /usr/local/cuda/lib64/libcublasLt.so
222222

223223
### COPY NATS & ETCD ###
224224
# Copy nats and etcd from dev image

0 commit comments

Comments
 (0)