Skip to content

Commit c516742

Browse files
authored
[docker] Fixes pytorch-gpu docker build script (#2139)
1 parent 4fafd87 commit c516742

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

serving/docker/pytorch-gpu.Dockerfile

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,10 +48,11 @@ COPY distribution[s]/ ./
4848
RUN mv *.deb djl-serving_all.deb || true
4949

5050
COPY scripts scripts/
51+
SHELL ["/bin/bash", "-c"]
5152
RUN chmod +x /usr/local/bin/dockerd-entrypoint.sh && \
5253
scripts/install_djl_serving.sh $djl_version && \
5354
scripts/install_djl_serving.sh $djl_version ${torch_version} && \
54-
djl-serving -i ai.djl.onnxruntime:onnxruntime-engine:$djl_version && \
55+
djl-serving -i ai.djl.onnxruntime:onnxruntime-engine:${djl_version//[~]/-} && \
5556
djl-serving -i com.microsoft.onnxruntime:onnxruntime_gpu:$onnx_version && \
5657
scripts/install_python.sh ${python_version} && \
5758
scripts/install_s5cmd.sh x64 && \

0 commit comments

Comments
 (0)