Skip to content

Commit 5a0dd00

Browse files
committed
Added spacy cpu&gpu
1 parent e064e54 commit 5a0dd00

4 files changed

Lines changed: 65 additions & 4 deletions

File tree

build.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ docker build --build-arg VERSION=$VERSION --build-arg CPU_GPU=gpu -t instituutne
2828
docker build --build-arg VERSION=$VERSION --build-arg CPU_GPU=cpu -t instituutnederlandsetaal/galahad-taggers-flair-cpu:$VERSION flair
2929
docker build --build-arg VERSION=$VERSION --build-arg CPU_GPU=gpu -t instituutnederlandsetaal/galahad-taggers-flair-gpu:$VERSION flair
3030
# # spacy
31-
#docker build --build-arg VERSION=$VERSION --build-arg SPACY_MODEL=nl_core_news_lg --build-arg CPU_GPU=cpu -t instituutnederlandsetaal/galahad-taggers-spacy-cpu:$VERSION spacy
32-
# docker build --build-arg VERSION=$VERSION --build-arg SPACY_MODEL=nl_core_news_lg --build-arg CPU_GPU=gpu -t instituutnederlandsetaal/galahad-taggers-spacy-gpu:$VERSION spacy
31+
docker build --build-arg VERSION=$VERSION --build-arg SPACY_MODEL=nl_core_news_lg --build-arg CPU_GPU=cpu -t instituutnederlandsetaal/galahad-taggers-spacy-cpu:$VERSION spacy
32+
docker build --build-arg VERSION=$VERSION --build-arg SPACY_MODEL=nl_core_news_lg --build-arg CPU_GPU=gpu -t instituutnederlandsetaal/galahad-taggers-spacy-gpu:$VERSION spacy
3333
# stanza
3434
docker build --build-arg VERSION=$VERSION --build-arg CPU_GPU=cpu -t instituutnederlandsetaal/galahad-taggers-stanza-cpu:$VERSION stanza
3535
docker build --build-arg VERSION=$VERSION --build-arg CPU_GPU=gpu -t instituutnederlandsetaal/galahad-taggers-stanza-gpu:$VERSION stanza

push-and-build.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ docker push instituutnederlandsetaal/galahad-taggers-pie-gpu-tdn-all:$VERSION
1515

1616
# UD-parsers
1717
# docker push instituutnederlandsetaal/galahad-taggers-udpipe:$VERSION
18-
# docker push instituutnederlandsetaal/galahad-taggers-spacy:$VERSION
18+
docker push instituutnederlandsetaal/galahad-taggers-spacy-cpu:$VERSION
19+
docker push instituutnederlandsetaal/galahad-taggers-spacy-gpu:$VERSION
1920
docker push instituutnederlandsetaal/galahad-taggers-stanza-cpu:$VERSION
2021
docker push instituutnederlandsetaal/galahad-taggers-stanza-gpu:$VERSION
2122
docker push instituutnederlandsetaal/galahad-taggers-flair-cpu:$VERSION

spacy/fix-cuda-env.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
export LD_LIBRARY_PATH=:/usr/local/lib/python3.10/dist-packages/nvidia/cuda_runtime/lib:/usr/local/lib/python3.10/dist-packages/nvidia/cublas/lib:/usr/local/lib/python3.10/dist-packages/nvidia/cufft/lib:/usr/local/lib/python3.10/dist-packages/nvidia/curand/lib:/usr/local/lib/python3.10/dist-packages/nvidia/cusparse/lib:/usr/local/lib/python3.10/dist-packages/nvidia/cuda_nvrtc/lib:/usr/local/lib/python3.10/dist-packages/nvidia/cusolver/lib
1+
export LD_LIBRARY_PATH=:/usr/local/lib/python3.12/dist-packages/nvidia/cuda_runtime/lib:/usr/local/lib/python3.12/dist-packages/nvidia/cublas/lib:/usr/local/lib/python3.12/dist-packages/nvidia/cufft/lib:/usr/local/lib/python3.12/dist-packages/nvidia/curand/lib:/usr/local/lib/python3.12/dist-packages/nvidia/cusparse/lib:/usr/local/lib/python3.12/dist-packages/nvidia/cuda_nvrtc/lib:/usr/local/lib/python3.12/dist-packages/nvidia/cusolver/lib

spacy/requirements/gpu.txt

Lines changed: 60 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,60 @@
1+
annotated-types==0.7.0
2+
blis==1.2.1
3+
catalogue==2.0.10
4+
certifi==2025.1.31
5+
charset-normalizer==3.4.1
6+
click==8.1.8
7+
cloudpathlib==0.21.0
8+
confection==0.1.5
9+
conllu==6.0.0
10+
cupy-cuda12x==12.3.0
11+
cymem==2.0.11
12+
fastrlock==0.8.3
13+
idna==3.10
14+
jinja2==3.1.6
15+
langcodes==3.5.0
16+
language-data==1.3.0
17+
marisa-trie==1.2.1
18+
markdown-it-py==3.0.0
19+
markupsafe==3.0.2
20+
mdurl==0.1.2
21+
murmurhash==1.0.12
22+
nl-core-news-lg @ https://github.com/explosion/spacy-models/releases/download/nl_core_news_lg-3.8.0/nl_core_news_lg-3.8.0-py3-none-any.whl
23+
numpy==1.26.4
24+
nvidia-cublas-cu12==12.6.4.1
25+
nvidia-cuda-cupti-cu12==12.6.80
26+
nvidia-cuda-nvrtc-cu12==12.6.77
27+
nvidia-cuda-runtime-cu12==12.6.77
28+
nvidia-cudnn-cu12==9.5.1.17
29+
nvidia-cufft-cu12==11.3.0.4
30+
nvidia-curand-cu12==10.3.7.77
31+
nvidia-cusolver-cu12==11.7.1.2
32+
nvidia-cusparse-cu12==12.5.4.2
33+
nvidia-nccl-cu12==2.21.5
34+
nvidia-nvjitlink-cu12==12.6.85
35+
nvidia-nvtx-cu12==12.6.77
36+
packaging==24.2
37+
pip==24.3.1
38+
preshed==3.0.9
39+
pydantic==2.11.1
40+
pydantic-core==2.33.0
41+
pygments==2.19.1
42+
requests==2.32.3
43+
rich==14.0.0
44+
setuptools==78.1.0
45+
shellingham==1.5.4
46+
smart-open==7.1.0
47+
spacy==3.8.5
48+
spacy-conll==4.0.1
49+
spacy-legacy==3.0.12
50+
spacy-loggers==1.0.5
51+
srsly==2.5.1
52+
thinc==8.3.4
53+
tqdm==4.67.1
54+
typer==0.15.2
55+
typing-extensions==4.13.0
56+
typing-inspection==0.4.0
57+
urllib3==2.3.0
58+
wasabi==1.1.3
59+
weasel==0.4.1
60+
wrapt==1.17.2

0 commit comments

Comments
 (0)