Skip to content

Commit 4be4271

Browse files
authored
Remove extra cupy installation
http://b/238238348
1 parent 6959e47 commit 4be4271

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
@@ -158,7 +158,7 @@ RUN pip install jax[cpu] && \
158158

159159
# Install spacy
160160
{{ if eq .Accelerator "gpu" }}
161-
RUN pip install spacy[cuda$CUDA_MAJOR_VERSION$CUDA_MINOR_VERSION] && \
161+
RUN mamba install -c conda-forge spacy cupy && \
162162
/tmp/clean-layer.sh
163163
{{ else }}
164164
RUN pip install spacy && \

0 commit comments

Comments
 (0)