File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -217,8 +217,8 @@ COPY --from=framework /usr/local/cuda/bin/fatbinary /usr/local/cuda/bin/fatbinar
217217COPY --from=framework /usr/local/cuda/include/ /usr/local/cuda/include/
218218COPY --from=framework /usr/local/cuda/nvvm /usr/local/cuda/nvvm
219219COPY --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
You can’t perform that action at this time.
0 commit comments