Skip to content

Commit 4813bd1

Browse files
authored
new parent image (#83)
1 parent 13790df commit 4813bd1

File tree

3 files changed

+59
-52
lines changed

3 files changed

+59
-52
lines changed

gpu-requirements.txt

Lines changed: 27 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@ aiohttp==3.8.5
1212
# openai
1313
aiosignal==1.3.1
1414
# via aiohttp
15-
anyio==4.0.0
15+
anyio==4.3.0
1616
# via
1717
# -r requirements/torch-cuda-requirements.txt
1818
# starlette
@@ -36,12 +36,12 @@ catalogue==2.0.8
3636
# spacy
3737
# srsly
3838
# thinc
39-
certifi==2023.7.22
39+
certifi==2024.2.2
4040
# via
4141
# -r requirements/torch-cuda-requirements.txt
4242
# minio
4343
# requests
44-
charset-normalizer==3.2.0
44+
charset-normalizer==3.3.2
4545
# via
4646
# -r requirements/torch-cuda-requirements.txt
4747
# aiohttp
@@ -63,13 +63,13 @@ cymem==2.0.7
6363
# thinc
6464
embedders==0.1.8
6565
# via -r requirements/gpu-requirements.in
66-
exceptiongroup==1.1.3
66+
exceptiongroup==1.2.0
6767
# via
6868
# -r requirements/torch-cuda-requirements.txt
6969
# anyio
70-
fastapi==0.95.2
70+
fastapi==0.110.0
7171
# via -r requirements/torch-cuda-requirements.txt
72-
filelock==3.12.4
72+
filelock==3.13.1
7373
# via
7474
# -r requirements/torch-cuda-requirements.txt
7575
# huggingface-hub
@@ -78,20 +78,21 @@ frozenlist==1.3.3
7878
# via
7979
# aiohttp
8080
# aiosignal
81-
fsspec==2023.9.1
81+
fsspec==2024.2.0
8282
# via
8383
# -r requirements/torch-cuda-requirements.txt
8484
# huggingface-hub
8585
h11==0.14.0
8686
# via
8787
# -r requirements/torch-cuda-requirements.txt
8888
# uvicorn
89-
huggingface-hub==0.17.2
89+
huggingface-hub==0.21.3
9090
# via
9191
# -r requirements/torch-cuda-requirements.txt
9292
# sentence-transformers
93+
# tokenizers
9394
# transformers
94-
idna==3.4
95+
idna==3.6
9596
# via
9697
# -r requirements/torch-cuda-requirements.txt
9798
# anyio
@@ -145,7 +146,7 @@ numpy==1.23.4
145146
# transformers
146147
openai==0.27.8
147148
# via embedders
148-
packaging==23.1
149+
packaging==23.2
149150
# via
150151
# -r requirements/torch-cuda-requirements.txt
151152
# huggingface-hub
@@ -161,9 +162,9 @@ preshed==3.0.8
161162
# via
162163
# spacy
163164
# thinc
164-
psycopg2-binary==2.9.4
165+
psycopg2-binary==2.9.9
165166
# via -r requirements/torch-cuda-requirements.txt
166-
pyaml==23.9.6
167+
pyaml==23.12.0
167168
# via
168169
# -r requirements/torch-cuda-requirements.txt
169170
# scikit-optimize
@@ -174,12 +175,12 @@ pydantic==1.10.8
174175
# fastapi
175176
# spacy
176177
# thinc
177-
python-dateutil==2.8.2
178+
python-dateutil==2.9.0.post0
178179
# via
179180
# -r requirements/torch-cuda-requirements.txt
180181
# botocore
181182
# pandas
182-
pytz==2023.3.post1
183+
pytz==2024.1
183184
# via
184185
# -r requirements/torch-cuda-requirements.txt
185186
# pandas
@@ -189,7 +190,7 @@ pyyaml==6.0.1
189190
# huggingface-hub
190191
# pyaml
191192
# transformers
192-
regex==2023.8.8
193+
regex==2023.12.25
193194
# via
194195
# -r requirements/torch-cuda-requirements.txt
195196
# nltk
@@ -207,7 +208,7 @@ s3transfer==0.6.2
207208
# via
208209
# -r requirements/torch-cuda-requirements.txt
209210
# boto3
210-
safetensors==0.3.3
211+
safetensors==0.4.2
211212
# via
212213
# -r requirements/torch-cuda-requirements.txt
213214
# transformers
@@ -219,7 +220,7 @@ scikit-learn==1.1.2
219220
# sentence-transformers
220221
scikit-optimize==0.9.0
221222
# via -r requirements/torch-cuda-requirements.txt
222-
scipy==1.11.2
223+
scipy==1.12.0
223224
# via
224225
# -r requirements/torch-cuda-requirements.txt
225226
# scikit-learn
@@ -235,7 +236,7 @@ six==1.16.0
235236
# python-dateutil
236237
smart-open==5.2.1
237238
# via pathy
238-
sniffio==1.3.0
239+
sniffio==1.3.1
239240
# via
240241
# -r requirements/torch-cuda-requirements.txt
241242
# anyio
@@ -254,17 +255,17 @@ srsly==2.4.5
254255
# confection
255256
# spacy
256257
# thinc
257-
starlette==0.27.0
258+
starlette==0.36.3
258259
# via
259260
# -r requirements/torch-cuda-requirements.txt
260261
# fastapi
261262
thinc==8.1.5
262263
# via spacy
263-
threadpoolctl==3.2.0
264+
threadpoolctl==3.3.0
264265
# via
265266
# -r requirements/torch-cuda-requirements.txt
266267
# scikit-learn
267-
tokenizers==0.13.3
268+
tokenizers==0.15.2
268269
# via
269270
# -r requirements/torch-cuda-requirements.txt
270271
# transformers
@@ -278,7 +279,7 @@ torchvision==0.14.1
278279
# via
279280
# -r requirements/gpu-requirements.in
280281
# sentence-transformers
281-
tqdm==4.66.1
282+
tqdm==4.66.2
282283
# via
283284
# -r requirements/torch-cuda-requirements.txt
284285
# embedders
@@ -288,7 +289,7 @@ tqdm==4.66.1
288289
# sentence-transformers
289290
# spacy
290291
# transformers
291-
transformers==4.30.0
292+
transformers==4.38.2
292293
# via
293294
# -r requirements/torch-cuda-requirements.txt
294295
# embedders
@@ -297,9 +298,11 @@ typer==0.4.2
297298
# via
298299
# pathy
299300
# spacy
300-
typing-extensions==4.8.0
301+
typing-extensions==4.10.0
301302
# via
302303
# -r requirements/torch-cuda-requirements.txt
304+
# anyio
305+
# fastapi
303306
# huggingface-hub
304307
# pydantic
305308
# starlette

gpu.Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM kernai/refinery-parent-images:v1.13.0-torch-cuda
1+
FROM kernai/refinery-parent-images:v1.14.0-torch-cuda
22

33
WORKDIR /program
44

requirements/torch-cuda-requirements.txt

Lines changed: 31 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,42 +1,44 @@
11
#
2-
# This file is autogenerated by pip-compile with python 3.9
3-
# To update, run:
2+
# This file is autogenerated by pip-compile with Python 3.9
3+
# by the following command:
44
#
55
# pip-compile torch-cuda-requirements.in
66
#
77
--extra-index-url https://download.pytorch.org/whl/cu113
88

9-
anyio==4.0.0
9+
anyio==4.3.0
1010
# via starlette
1111
boto3==1.25.0
1212
# via -r common-requirements.in
1313
botocore==1.28.5
1414
# via
1515
# boto3
1616
# s3transfer
17-
certifi==2023.7.22
17+
certifi==2024.2.2
1818
# via
1919
# minio
2020
# requests
21-
charset-normalizer==3.2.0
21+
charset-normalizer==3.3.2
2222
# via requests
2323
click==8.1.7
2424
# via uvicorn
25-
exceptiongroup==1.1.3
25+
exceptiongroup==1.2.0
2626
# via anyio
27-
fastapi==0.95.2
27+
fastapi==0.110.0
2828
# via -r mini-requirements.in
29-
filelock==3.12.4
29+
filelock==3.13.1
3030
# via
3131
# huggingface-hub
3232
# transformers
33-
fsspec==2023.9.1
33+
fsspec==2024.2.0
3434
# via huggingface-hub
3535
h11==0.14.0
3636
# via uvicorn
37-
huggingface-hub==0.17.2
38-
# via transformers
39-
idna==3.4
37+
huggingface-hub==0.21.3
38+
# via
39+
# tokenizers
40+
# transformers
41+
idna==3.6
4042
# via
4143
# anyio
4244
# requests
@@ -58,32 +60,32 @@ numpy==1.23.4
5860
# scikit-optimize
5961
# scipy
6062
# transformers
61-
packaging==23.1
63+
packaging==23.2
6264
# via
6365
# huggingface-hub
6466
# transformers
6567
pandas==1.5.1
6668
# via -r common-requirements.in
67-
psycopg2-binary==2.9.4
69+
psycopg2-binary==2.9.9
6870
# via -r common-requirements.in
69-
pyaml==23.9.6
71+
pyaml==23.12.0
7072
# via scikit-optimize
7173
pydantic==1.10.8
7274
# via
7375
# -r mini-requirements.in
7476
# fastapi
75-
python-dateutil==2.8.2
77+
python-dateutil==2.9.0.post0
7678
# via
7779
# botocore
7880
# pandas
79-
pytz==2023.3.post1
81+
pytz==2024.1
8082
# via pandas
8183
pyyaml==6.0.1
8284
# via
8385
# huggingface-hub
8486
# pyaml
8587
# transformers
86-
regex==2023.8.8
88+
regex==2023.12.25
8789
# via transformers
8890
requests==2.31.0
8991
# via
@@ -92,40 +94,42 @@ requests==2.31.0
9294
# transformers
9395
s3transfer==0.6.2
9496
# via boto3
95-
safetensors==0.3.3
97+
safetensors==0.4.2
9698
# via transformers
9799
scikit-learn==1.1.2
98100
# via
99101
# -r torch-cuda-requirements.in
100102
# scikit-optimize
101103
scikit-optimize==0.9.0
102104
# via -r torch-cuda-requirements.in
103-
scipy==1.11.2
105+
scipy==1.12.0
104106
# via
105107
# scikit-learn
106108
# scikit-optimize
107109
six==1.16.0
108110
# via python-dateutil
109-
sniffio==1.3.0
111+
sniffio==1.3.1
110112
# via anyio
111113
sqlalchemy==1.4.42
112114
# via -r common-requirements.in
113-
starlette==0.27.0
115+
starlette==0.36.3
114116
# via fastapi
115-
threadpoolctl==3.2.0
117+
threadpoolctl==3.3.0
116118
# via scikit-learn
117-
tokenizers==0.13.3
119+
tokenizers==0.15.2
118120
# via transformers
119121
torch==1.13.1
120122
# via -r torch-cuda-requirements.in
121-
tqdm==4.66.1
123+
tqdm==4.66.2
122124
# via
123125
# huggingface-hub
124126
# transformers
125-
transformers==4.30.0
127+
transformers==4.38.2
126128
# via -r torch-cuda-requirements.in
127-
typing-extensions==4.8.0
129+
typing-extensions==4.10.0
128130
# via
131+
# anyio
132+
# fastapi
129133
# huggingface-hub
130134
# pydantic
131135
# starlette

0 commit comments

Comments
 (0)