Force cpu-only torch

Author: damccorm

sdks/python/container/Dockerfile

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

sdks/python/container/common.gradle

@@ -53,7 +53,7 @@ def generatePythonRequirements = tasks.register("generatePythonRequirements") {
               "${files(configurations.sdkSourceTarball.files).singleFile} " +
               "ml_image_requirements.txt " +
               "container/ml " +
-              "[gcp,dataframe,test,tensorflow,torch,transformers] " +
+              "[gcp,dataframe,test,ml_cpu] " +
               "${pipExtraOptions}"
     }
   }

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

@@ -45,7 +45,7 @@ click==8.3.0
 cloud-sql-python-connector==1.18.5
 crcmod==1.7
 cryptography==46.0.3
-Cython==3.1.4
+Cython==3.1.5
 dill==0.3.1.1
 dnspython==2.8.0
 docker==7.1.0
@@ -79,17 +79,17 @@ google-cloud-profiler==4.1.0
 google-cloud-pubsub==2.31.1
 google-cloud-pubsublite==1.12.0
 google-cloud-recommendations-ai==0.10.18
-google-cloud-resource-manager==1.14.2
+google-cloud-resource-manager==1.15.0
 google-cloud-secret-manager==2.25.0
 google-cloud-spanner==3.58.0
 google-cloud-storage==2.19.0
-google-cloud-videointelligence==2.16.2
-google-cloud-vision==3.10.2
+google-cloud-videointelligence==2.17.0
+google-cloud-vision==3.11.0
 google-crc32c==1.7.1
 google-genai==1.45.0
 google-pasta==0.2.0
 google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
+googleapis-common-protos==1.71.0
 greenlet==3.2.4
 grpc-google-iam-v1==0.14.3
 grpc-interceptor==0.15.4
@@ -104,7 +104,7 @@ httpcore==1.0.9
 httplib2==0.22.0
 httpx==0.28.1
 huggingface-hub==0.35.3
-hypothesis==6.142.1
+hypothesis==6.142.2
 idna==3.11
 importlib_metadata==8.7.0
 iniconfig==2.3.0
@@ -136,20 +136,6 @@ namex==0.1.0
 networkx==3.4.2
 nltk==3.9.2
 numpy==2.2.6
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.5.1.17
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.26.2
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
 oauth2client==4.1.3
 objsize==0.7.1
 opentelemetry-api==1.38.0
@@ -197,7 +183,7 @@ pytz==2025.2
 PyYAML==6.0.3
 redis==5.3.1
 referencing==0.37.0
-regex==2025.9.18
+regex==2025.10.23
 requests==2.32.5
 requests-mock==1.12.1
 rich==14.2.0
@@ -228,10 +214,9 @@ testcontainers==4.13.2
 threadpoolctl==3.6.0
 tokenizers==0.21.4
 tomli==2.3.0
-torch==2.7.1
+torch==2.8.0+cpu
 tqdm==4.67.1
 transformers==4.55.4
-triton==3.3.1
 typing-inspection==0.4.2
 typing_extensions==4.15.0
 tzdata==2025.2
@@ -242,7 +227,7 @@ virtualenv-clone==0.5.7
 websockets==15.0.1
 Werkzeug==3.1.3
 wheel==0.45.1
-wrapt==1.17.3
+wrapt==2.0.0
 yarl==1.22.0
 zipp==3.23.0
 zstandard==0.25.0

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

@@ -44,7 +44,7 @@ click==8.3.0
 cloud-sql-python-connector==1.18.5
 crcmod==1.7
 cryptography==46.0.3
-Cython==3.1.4
+Cython==3.1.5
 dill==0.3.1.1
 dnspython==2.8.0
 docker==7.1.0
@@ -77,17 +77,17 @@ google-cloud-profiler==4.1.0
 google-cloud-pubsub==2.31.1
 google-cloud-pubsublite==1.12.0
 google-cloud-recommendations-ai==0.10.18
-google-cloud-resource-manager==1.14.2
+google-cloud-resource-manager==1.15.0
 google-cloud-secret-manager==2.25.0
 google-cloud-spanner==3.58.0
 google-cloud-storage==2.19.0
-google-cloud-videointelligence==2.16.2
-google-cloud-vision==3.10.2
+google-cloud-videointelligence==2.17.0
+google-cloud-vision==3.11.0
 google-crc32c==1.7.1
 google-genai==1.45.0
 google-pasta==0.2.0
 google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
+googleapis-common-protos==1.71.0
 greenlet==3.2.4
 grpc-google-iam-v1==0.14.3
 grpc-interceptor==0.15.4
@@ -102,7 +102,7 @@ httpcore==1.0.9
 httplib2==0.22.0
 httpx==0.28.1
 huggingface-hub==0.35.3
-hypothesis==6.142.1
+hypothesis==6.142.2
 idna==3.11
 importlib_metadata==8.7.0
 iniconfig==2.3.0
@@ -134,20 +134,6 @@ namex==0.1.0
 networkx==3.5
 nltk==3.9.2
 numpy==2.2.6
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.5.1.17
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.26.2
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
 oauth2client==4.1.3
 objsize==0.7.1
 opentelemetry-api==1.38.0
@@ -195,7 +181,7 @@ pytz==2025.2
 PyYAML==6.0.3
 redis==5.3.1
 referencing==0.37.0
-regex==2025.9.18
+regex==2025.10.23
 requests==2.32.5
 requests-mock==1.12.1
 rich==14.2.0
@@ -225,10 +211,9 @@ termcolor==3.1.0
 testcontainers==4.13.2
 threadpoolctl==3.6.0
 tokenizers==0.21.4
-torch==2.7.1
+torch==2.8.0+cpu
 tqdm==4.67.1
 transformers==4.55.4
-triton==3.3.1
 typing-inspection==0.4.2
 typing_extensions==4.15.0
 tzdata==2025.2
@@ -239,7 +224,7 @@ virtualenv-clone==0.5.7
 websockets==15.0.1
 Werkzeug==3.1.3
 wheel==0.45.1
-wrapt==1.17.3
+wrapt==2.0.0
 yarl==1.22.0
 zipp==3.23.0
 zstandard==0.25.0

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

@@ -43,7 +43,7 @@ click==8.3.0
 cloud-sql-python-connector==1.18.5
 crcmod==1.7
 cryptography==46.0.3
-Cython==3.1.4
+Cython==3.1.5
 dill==0.3.1.1
 dnspython==2.8.0
 docker==7.1.0
@@ -76,17 +76,17 @@ google-cloud-profiler==4.1.0
 google-cloud-pubsub==2.31.1
 google-cloud-pubsublite==1.12.0
 google-cloud-recommendations-ai==0.10.18
-google-cloud-resource-manager==1.14.2
+google-cloud-resource-manager==1.15.0
 google-cloud-secret-manager==2.25.0
 google-cloud-spanner==3.58.0
 google-cloud-storage==2.19.0
-google-cloud-videointelligence==2.16.2
-google-cloud-vision==3.10.2
+google-cloud-videointelligence==2.17.0
+google-cloud-vision==3.11.0
 google-crc32c==1.7.1
 google-genai==1.45.0
 google-pasta==0.2.0
 google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
+googleapis-common-protos==1.71.0
 greenlet==3.2.4
 grpc-google-iam-v1==0.14.3
 grpc-interceptor==0.15.4
@@ -101,7 +101,7 @@ httpcore==1.0.9
 httplib2==0.22.0
 httpx==0.28.1
 huggingface-hub==0.35.3
-hypothesis==6.142.1
+hypothesis==6.142.2
 idna==3.11
 importlib_metadata==8.7.0
 iniconfig==2.3.0
@@ -133,20 +133,6 @@ namex==0.1.0
 networkx==3.5
 nltk==3.9.2
 numpy==2.2.6
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.5.1.17
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.26.2
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
 oauth2client==4.1.3
 objsize==0.7.1
 opentelemetry-api==1.38.0
@@ -194,7 +180,7 @@ pytz==2025.2
 PyYAML==6.0.3
 redis==5.3.1
 referencing==0.37.0
-regex==2025.9.18
+regex==2025.10.23
 requests==2.32.5
 requests-mock==1.12.1
 rich==14.2.0
@@ -224,10 +210,9 @@ termcolor==3.1.0
 testcontainers==4.13.2
 threadpoolctl==3.6.0
 tokenizers==0.21.4
-torch==2.7.1
+torch==2.8.0+cpu
 tqdm==4.67.1
 transformers==4.55.4
-triton==3.3.1
 typing-inspection==0.4.2
 typing_extensions==4.15.0
 tzdata==2025.2
@@ -238,7 +223,7 @@ virtualenv-clone==0.5.7
 websockets==15.0.1
 Werkzeug==3.1.3
 wheel==0.45.1
-wrapt==1.17.3
+wrapt==2.0.0
 yarl==1.22.0
 zipp==3.23.0
 zstandard==0.25.0

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

@@ -43,7 +43,7 @@ click==8.3.0
 cloud-sql-python-connector==1.18.5
 crcmod==1.7
 cryptography==46.0.3
-Cython==3.1.4
+Cython==3.1.5
 dill==0.3.1.1
 dnspython==2.8.0
 docker==7.1.0
@@ -74,22 +74,22 @@ google-cloud-language==2.18.0
 google-cloud-pubsub==2.31.1
 google-cloud-pubsublite==1.12.0
 google-cloud-recommendations-ai==0.10.18
-google-cloud-resource-manager==1.14.2
+google-cloud-resource-manager==1.15.0
 google-cloud-secret-manager==2.25.0
 google-cloud-spanner==3.58.0
 google-cloud-storage==2.19.0
-google-cloud-videointelligence==2.16.2
-google-cloud-vision==3.10.2
+google-cloud-videointelligence==2.17.0
+google-cloud-vision==3.11.0
 google-crc32c==1.7.1
 google-genai==1.45.0
 google-pasta==0.2.0
 google-resumable-media==2.7.2
-googleapis-common-protos==1.70.0
+googleapis-common-protos==1.71.0
 greenlet==3.2.4
 grpc-google-iam-v1==0.14.3
 grpc-interceptor==0.15.4
-grpcio==1.76.0rc1
-grpcio-status==1.76.0rc1
+grpcio==1.76.0
+grpcio-status==1.76.0
 guppy3==3.1.5
 h11==0.16.0
 h5py==3.15.1
@@ -99,7 +99,7 @@ httpcore==1.0.9
 httplib2==0.22.0
 httpx==0.28.1
 huggingface-hub==0.35.3
-hypothesis==6.142.1
+hypothesis==6.142.2
 idna==3.11
 importlib_metadata==8.7.0
 iniconfig==2.3.0
@@ -130,20 +130,6 @@ namex==0.1.0
 networkx==3.5
 nltk==3.9.2
 numpy==2.2.6
-nvidia-cublas-cu12==12.6.4.1
-nvidia-cuda-cupti-cu12==12.6.80
-nvidia-cuda-nvrtc-cu12==12.6.77
-nvidia-cuda-runtime-cu12==12.6.77
-nvidia-cudnn-cu12==9.5.1.17
-nvidia-cufft-cu12==11.3.0.4
-nvidia-cufile-cu12==1.11.1.6
-nvidia-curand-cu12==10.3.7.77
-nvidia-cusolver-cu12==11.7.1.2
-nvidia-cusparse-cu12==12.5.4.2
-nvidia-cusparselt-cu12==0.6.3
-nvidia-nccl-cu12==2.26.2
-nvidia-nvjitlink-cu12==12.6.85
-nvidia-nvtx-cu12==12.6.77
 oauth2client==4.1.3
 objsize==0.7.1
 opentelemetry-api==1.38.0
@@ -191,7 +177,7 @@ pytz==2025.2
 PyYAML==6.0.3
 redis==5.3.1
 referencing==0.37.0
-regex==2025.9.18
+regex==2025.10.23
 requests==2.32.5
 requests-mock==1.12.1
 rich==14.2.0
@@ -221,10 +207,9 @@ termcolor==3.1.0
 testcontainers==4.13.2
 threadpoolctl==3.6.0
 tokenizers==0.21.4
-torch==2.7.1
+torch==2.8.0+cpu
 tqdm==4.67.1
 transformers==4.55.4
-triton==3.3.1
 typing-inspection==0.4.2
 typing_extensions==4.15.0
 tzdata==2025.2
@@ -234,7 +219,7 @@ virtualenv-clone==0.5.7
 websockets==15.0.1
 Werkzeug==3.1.3
 wheel==0.45.1
-wrapt==1.17.3
+wrapt==2.0.0
 yarl==1.22.0
 zipp==3.23.0
 zstandard==0.25.0