Skip to content

Commit 8815fab

Browse files
committed
Force cpu-only torch
1 parent 21cd6a8 commit 8815fab

File tree

9 files changed

+58
-126
lines changed

9 files changed

+58
-126
lines changed

sdks/python/container/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ RUN \
5050

5151
# Install required packages for Beam Python SDK and common dependencies used by users.
5252
# use --no-deps to ensure the list includes all transitive dependencies.
53-
pip install --no-deps -r /tmp/base_image_requirements.txt && \
53+
pip install --no-deps -r /tmp/base_image_requirements.txt --extra-index-url https://download.pytorch.org/whl/cpu && \
5454
rm -rf /tmp/base_image_requirements.txt && \
5555
python -c "import nltk; nltk.download('stopwords')" && \
5656
rm /root/nltk_data/corpora/stopwords.zip && \

sdks/python/container/common.gradle

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,7 @@ def generatePythonRequirements = tasks.register("generatePythonRequirements") {
5353
"${files(configurations.sdkSourceTarball.files).singleFile} " +
5454
"ml_image_requirements.txt " +
5555
"container/ml " +
56-
"[gcp,dataframe,test,tensorflow,torch,transformers] " +
56+
"[gcp,dataframe,test,ml_cpu] " +
5757
"${pipExtraOptions}"
5858
}
5959
}

sdks/python/container/ml/py310/ml_image_requirements.txt

Lines changed: 9 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ click==8.3.0
4545
cloud-sql-python-connector==1.18.5
4646
crcmod==1.7
4747
cryptography==46.0.3
48-
Cython==3.1.4
48+
Cython==3.1.5
4949
dill==0.3.1.1
5050
dnspython==2.8.0
5151
docker==7.1.0
@@ -79,17 +79,17 @@ google-cloud-profiler==4.1.0
7979
google-cloud-pubsub==2.31.1
8080
google-cloud-pubsublite==1.12.0
8181
google-cloud-recommendations-ai==0.10.18
82-
google-cloud-resource-manager==1.14.2
82+
google-cloud-resource-manager==1.15.0
8383
google-cloud-secret-manager==2.25.0
8484
google-cloud-spanner==3.58.0
8585
google-cloud-storage==2.19.0
86-
google-cloud-videointelligence==2.16.2
87-
google-cloud-vision==3.10.2
86+
google-cloud-videointelligence==2.17.0
87+
google-cloud-vision==3.11.0
8888
google-crc32c==1.7.1
8989
google-genai==1.45.0
9090
google-pasta==0.2.0
9191
google-resumable-media==2.7.2
92-
googleapis-common-protos==1.70.0
92+
googleapis-common-protos==1.71.0
9393
greenlet==3.2.4
9494
grpc-google-iam-v1==0.14.3
9595
grpc-interceptor==0.15.4
@@ -104,7 +104,7 @@ httpcore==1.0.9
104104
httplib2==0.22.0
105105
httpx==0.28.1
106106
huggingface-hub==0.35.3
107-
hypothesis==6.142.1
107+
hypothesis==6.142.2
108108
idna==3.11
109109
importlib_metadata==8.7.0
110110
iniconfig==2.3.0
@@ -136,20 +136,6 @@ namex==0.1.0
136136
networkx==3.4.2
137137
nltk==3.9.2
138138
numpy==2.2.6
139-
nvidia-cublas-cu12==12.6.4.1
140-
nvidia-cuda-cupti-cu12==12.6.80
141-
nvidia-cuda-nvrtc-cu12==12.6.77
142-
nvidia-cuda-runtime-cu12==12.6.77
143-
nvidia-cudnn-cu12==9.5.1.17
144-
nvidia-cufft-cu12==11.3.0.4
145-
nvidia-cufile-cu12==1.11.1.6
146-
nvidia-curand-cu12==10.3.7.77
147-
nvidia-cusolver-cu12==11.7.1.2
148-
nvidia-cusparse-cu12==12.5.4.2
149-
nvidia-cusparselt-cu12==0.6.3
150-
nvidia-nccl-cu12==2.26.2
151-
nvidia-nvjitlink-cu12==12.6.85
152-
nvidia-nvtx-cu12==12.6.77
153139
oauth2client==4.1.3
154140
objsize==0.7.1
155141
opentelemetry-api==1.38.0
@@ -197,7 +183,7 @@ pytz==2025.2
197183
PyYAML==6.0.3
198184
redis==5.3.1
199185
referencing==0.37.0
200-
regex==2025.9.18
186+
regex==2025.10.23
201187
requests==2.32.5
202188
requests-mock==1.12.1
203189
rich==14.2.0
@@ -228,10 +214,9 @@ testcontainers==4.13.2
228214
threadpoolctl==3.6.0
229215
tokenizers==0.21.4
230216
tomli==2.3.0
231-
torch==2.7.1
217+
torch==2.8.0+cpu
232218
tqdm==4.67.1
233219
transformers==4.55.4
234-
triton==3.3.1
235220
typing-inspection==0.4.2
236221
typing_extensions==4.15.0
237222
tzdata==2025.2
@@ -242,7 +227,7 @@ virtualenv-clone==0.5.7
242227
websockets==15.0.1
243228
Werkzeug==3.1.3
244229
wheel==0.45.1
245-
wrapt==1.17.3
230+
wrapt==2.0.0
246231
yarl==1.22.0
247232
zipp==3.23.0
248233
zstandard==0.25.0

sdks/python/container/ml/py311/ml_image_requirements.txt

Lines changed: 9 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ click==8.3.0
4444
cloud-sql-python-connector==1.18.5
4545
crcmod==1.7
4646
cryptography==46.0.3
47-
Cython==3.1.4
47+
Cython==3.1.5
4848
dill==0.3.1.1
4949
dnspython==2.8.0
5050
docker==7.1.0
@@ -77,17 +77,17 @@ google-cloud-profiler==4.1.0
7777
google-cloud-pubsub==2.31.1
7878
google-cloud-pubsublite==1.12.0
7979
google-cloud-recommendations-ai==0.10.18
80-
google-cloud-resource-manager==1.14.2
80+
google-cloud-resource-manager==1.15.0
8181
google-cloud-secret-manager==2.25.0
8282
google-cloud-spanner==3.58.0
8383
google-cloud-storage==2.19.0
84-
google-cloud-videointelligence==2.16.2
85-
google-cloud-vision==3.10.2
84+
google-cloud-videointelligence==2.17.0
85+
google-cloud-vision==3.11.0
8686
google-crc32c==1.7.1
8787
google-genai==1.45.0
8888
google-pasta==0.2.0
8989
google-resumable-media==2.7.2
90-
googleapis-common-protos==1.70.0
90+
googleapis-common-protos==1.71.0
9191
greenlet==3.2.4
9292
grpc-google-iam-v1==0.14.3
9393
grpc-interceptor==0.15.4
@@ -102,7 +102,7 @@ httpcore==1.0.9
102102
httplib2==0.22.0
103103
httpx==0.28.1
104104
huggingface-hub==0.35.3
105-
hypothesis==6.142.1
105+
hypothesis==6.142.2
106106
idna==3.11
107107
importlib_metadata==8.7.0
108108
iniconfig==2.3.0
@@ -134,20 +134,6 @@ namex==0.1.0
134134
networkx==3.5
135135
nltk==3.9.2
136136
numpy==2.2.6
137-
nvidia-cublas-cu12==12.6.4.1
138-
nvidia-cuda-cupti-cu12==12.6.80
139-
nvidia-cuda-nvrtc-cu12==12.6.77
140-
nvidia-cuda-runtime-cu12==12.6.77
141-
nvidia-cudnn-cu12==9.5.1.17
142-
nvidia-cufft-cu12==11.3.0.4
143-
nvidia-cufile-cu12==1.11.1.6
144-
nvidia-curand-cu12==10.3.7.77
145-
nvidia-cusolver-cu12==11.7.1.2
146-
nvidia-cusparse-cu12==12.5.4.2
147-
nvidia-cusparselt-cu12==0.6.3
148-
nvidia-nccl-cu12==2.26.2
149-
nvidia-nvjitlink-cu12==12.6.85
150-
nvidia-nvtx-cu12==12.6.77
151137
oauth2client==4.1.3
152138
objsize==0.7.1
153139
opentelemetry-api==1.38.0
@@ -195,7 +181,7 @@ pytz==2025.2
195181
PyYAML==6.0.3
196182
redis==5.3.1
197183
referencing==0.37.0
198-
regex==2025.9.18
184+
regex==2025.10.23
199185
requests==2.32.5
200186
requests-mock==1.12.1
201187
rich==14.2.0
@@ -225,10 +211,9 @@ termcolor==3.1.0
225211
testcontainers==4.13.2
226212
threadpoolctl==3.6.0
227213
tokenizers==0.21.4
228-
torch==2.7.1
214+
torch==2.8.0+cpu
229215
tqdm==4.67.1
230216
transformers==4.55.4
231-
triton==3.3.1
232217
typing-inspection==0.4.2
233218
typing_extensions==4.15.0
234219
tzdata==2025.2
@@ -239,7 +224,7 @@ virtualenv-clone==0.5.7
239224
websockets==15.0.1
240225
Werkzeug==3.1.3
241226
wheel==0.45.1
242-
wrapt==1.17.3
227+
wrapt==2.0.0
243228
yarl==1.22.0
244229
zipp==3.23.0
245230
zstandard==0.25.0

sdks/python/container/ml/py312/ml_image_requirements.txt

Lines changed: 9 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ click==8.3.0
4343
cloud-sql-python-connector==1.18.5
4444
crcmod==1.7
4545
cryptography==46.0.3
46-
Cython==3.1.4
46+
Cython==3.1.5
4747
dill==0.3.1.1
4848
dnspython==2.8.0
4949
docker==7.1.0
@@ -76,17 +76,17 @@ google-cloud-profiler==4.1.0
7676
google-cloud-pubsub==2.31.1
7777
google-cloud-pubsublite==1.12.0
7878
google-cloud-recommendations-ai==0.10.18
79-
google-cloud-resource-manager==1.14.2
79+
google-cloud-resource-manager==1.15.0
8080
google-cloud-secret-manager==2.25.0
8181
google-cloud-spanner==3.58.0
8282
google-cloud-storage==2.19.0
83-
google-cloud-videointelligence==2.16.2
84-
google-cloud-vision==3.10.2
83+
google-cloud-videointelligence==2.17.0
84+
google-cloud-vision==3.11.0
8585
google-crc32c==1.7.1
8686
google-genai==1.45.0
8787
google-pasta==0.2.0
8888
google-resumable-media==2.7.2
89-
googleapis-common-protos==1.70.0
89+
googleapis-common-protos==1.71.0
9090
greenlet==3.2.4
9191
grpc-google-iam-v1==0.14.3
9292
grpc-interceptor==0.15.4
@@ -101,7 +101,7 @@ httpcore==1.0.9
101101
httplib2==0.22.0
102102
httpx==0.28.1
103103
huggingface-hub==0.35.3
104-
hypothesis==6.142.1
104+
hypothesis==6.142.2
105105
idna==3.11
106106
importlib_metadata==8.7.0
107107
iniconfig==2.3.0
@@ -133,20 +133,6 @@ namex==0.1.0
133133
networkx==3.5
134134
nltk==3.9.2
135135
numpy==2.2.6
136-
nvidia-cublas-cu12==12.6.4.1
137-
nvidia-cuda-cupti-cu12==12.6.80
138-
nvidia-cuda-nvrtc-cu12==12.6.77
139-
nvidia-cuda-runtime-cu12==12.6.77
140-
nvidia-cudnn-cu12==9.5.1.17
141-
nvidia-cufft-cu12==11.3.0.4
142-
nvidia-cufile-cu12==1.11.1.6
143-
nvidia-curand-cu12==10.3.7.77
144-
nvidia-cusolver-cu12==11.7.1.2
145-
nvidia-cusparse-cu12==12.5.4.2
146-
nvidia-cusparselt-cu12==0.6.3
147-
nvidia-nccl-cu12==2.26.2
148-
nvidia-nvjitlink-cu12==12.6.85
149-
nvidia-nvtx-cu12==12.6.77
150136
oauth2client==4.1.3
151137
objsize==0.7.1
152138
opentelemetry-api==1.38.0
@@ -194,7 +180,7 @@ pytz==2025.2
194180
PyYAML==6.0.3
195181
redis==5.3.1
196182
referencing==0.37.0
197-
regex==2025.9.18
183+
regex==2025.10.23
198184
requests==2.32.5
199185
requests-mock==1.12.1
200186
rich==14.2.0
@@ -224,10 +210,9 @@ termcolor==3.1.0
224210
testcontainers==4.13.2
225211
threadpoolctl==3.6.0
226212
tokenizers==0.21.4
227-
torch==2.7.1
213+
torch==2.8.0+cpu
228214
tqdm==4.67.1
229215
transformers==4.55.4
230-
triton==3.3.1
231216
typing-inspection==0.4.2
232217
typing_extensions==4.15.0
233218
tzdata==2025.2
@@ -238,7 +223,7 @@ virtualenv-clone==0.5.7
238223
websockets==15.0.1
239224
Werkzeug==3.1.3
240225
wheel==0.45.1
241-
wrapt==1.17.3
226+
wrapt==2.0.0
242227
yarl==1.22.0
243228
zipp==3.23.0
244229
zstandard==0.25.0

sdks/python/container/ml/py313/ml_image_requirements.txt

Lines changed: 11 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ click==8.3.0
4343
cloud-sql-python-connector==1.18.5
4444
crcmod==1.7
4545
cryptography==46.0.3
46-
Cython==3.1.4
46+
Cython==3.1.5
4747
dill==0.3.1.1
4848
dnspython==2.8.0
4949
docker==7.1.0
@@ -74,22 +74,22 @@ google-cloud-language==2.18.0
7474
google-cloud-pubsub==2.31.1
7575
google-cloud-pubsublite==1.12.0
7676
google-cloud-recommendations-ai==0.10.18
77-
google-cloud-resource-manager==1.14.2
77+
google-cloud-resource-manager==1.15.0
7878
google-cloud-secret-manager==2.25.0
7979
google-cloud-spanner==3.58.0
8080
google-cloud-storage==2.19.0
81-
google-cloud-videointelligence==2.16.2
82-
google-cloud-vision==3.10.2
81+
google-cloud-videointelligence==2.17.0
82+
google-cloud-vision==3.11.0
8383
google-crc32c==1.7.1
8484
google-genai==1.45.0
8585
google-pasta==0.2.0
8686
google-resumable-media==2.7.2
87-
googleapis-common-protos==1.70.0
87+
googleapis-common-protos==1.71.0
8888
greenlet==3.2.4
8989
grpc-google-iam-v1==0.14.3
9090
grpc-interceptor==0.15.4
91-
grpcio==1.76.0rc1
92-
grpcio-status==1.76.0rc1
91+
grpcio==1.76.0
92+
grpcio-status==1.76.0
9393
guppy3==3.1.5
9494
h11==0.16.0
9595
h5py==3.15.1
@@ -99,7 +99,7 @@ httpcore==1.0.9
9999
httplib2==0.22.0
100100
httpx==0.28.1
101101
huggingface-hub==0.35.3
102-
hypothesis==6.142.1
102+
hypothesis==6.142.2
103103
idna==3.11
104104
importlib_metadata==8.7.0
105105
iniconfig==2.3.0
@@ -130,20 +130,6 @@ namex==0.1.0
130130
networkx==3.5
131131
nltk==3.9.2
132132
numpy==2.2.6
133-
nvidia-cublas-cu12==12.6.4.1
134-
nvidia-cuda-cupti-cu12==12.6.80
135-
nvidia-cuda-nvrtc-cu12==12.6.77
136-
nvidia-cuda-runtime-cu12==12.6.77
137-
nvidia-cudnn-cu12==9.5.1.17
138-
nvidia-cufft-cu12==11.3.0.4
139-
nvidia-cufile-cu12==1.11.1.6
140-
nvidia-curand-cu12==10.3.7.77
141-
nvidia-cusolver-cu12==11.7.1.2
142-
nvidia-cusparse-cu12==12.5.4.2
143-
nvidia-cusparselt-cu12==0.6.3
144-
nvidia-nccl-cu12==2.26.2
145-
nvidia-nvjitlink-cu12==12.6.85
146-
nvidia-nvtx-cu12==12.6.77
147133
oauth2client==4.1.3
148134
objsize==0.7.1
149135
opentelemetry-api==1.38.0
@@ -191,7 +177,7 @@ pytz==2025.2
191177
PyYAML==6.0.3
192178
redis==5.3.1
193179
referencing==0.37.0
194-
regex==2025.9.18
180+
regex==2025.10.23
195181
requests==2.32.5
196182
requests-mock==1.12.1
197183
rich==14.2.0
@@ -221,10 +207,9 @@ termcolor==3.1.0
221207
testcontainers==4.13.2
222208
threadpoolctl==3.6.0
223209
tokenizers==0.21.4
224-
torch==2.7.1
210+
torch==2.8.0+cpu
225211
tqdm==4.67.1
226212
transformers==4.55.4
227-
triton==3.3.1
228213
typing-inspection==0.4.2
229214
typing_extensions==4.15.0
230215
tzdata==2025.2
@@ -234,7 +219,7 @@ virtualenv-clone==0.5.7
234219
websockets==15.0.1
235220
Werkzeug==3.1.3
236221
wheel==0.45.1
237-
wrapt==1.17.3
222+
wrapt==2.0.0
238223
yarl==1.22.0
239224
zipp==3.23.0
240225
zstandard==0.25.0

0 commit comments

Comments
 (0)