Merge pull request #2195 from Sefaria/up_cuda_12

chore(Dockerfile): upgrade spacy cuda
Sefaria · Dec 19, 2024 · 0a18102 · 0a18102
2 parents 16a8ada + 220e0ea
commit 0a18102
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/build/linker/Dockerfile b/build/linker/Dockerfile
@@ -46,5 +46,5 @@ ENV PATH=/usr/local/nvidia/bin:$PATH
 ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib64/
 ENV LD_PRELOAD=/usr/local/nvidia/lib64/libnvidia-ml.so
 
-RUN pip install spacy[cuda116]~=3.7.0
+RUN pip install spacy[cuda122]~=3.7.0