Skip to content

Commit ba6160c

Browse files
authored
Update Dockerfile.tmpl
1 parent 74046c2 commit ba6160c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Dockerfile.tmpl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -141,7 +141,7 @@ RUN pip install jax[cpu] && \
141141
# Install mxnet
142142
{{ if eq .Accelerator "gpu" }}
143143
# No specific package for 11.3 minor versions, using 11.2 instead.
144-
RUN pip install mxnet-cu$112 && \
144+
RUN pip install mxnet-cu112 && \
145145
/tmp/clean-layer.sh
146146
{{ else }}
147147
RUN pip install mxnet && \

0 commit comments

Comments
 (0)