chore(deps): update dependency torch to v2 (GoogleCloudPlatform#9287)

renovate-bot · web-flow · commit acd389448c99 · 2023-06-12T22:26:12.000Z
[![Mend Renovate](https://app.renovatebot.com/images/banner.svg)](https://renovatebot.com) This PR contains the following updates: | Package | Change | Age | Adoption | Passing | Confidence | |---|---|---|---|---|---| | [torch](https://pytorch.org/) | `==1.13.1` -> `==2.0.0` | [![age](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/age-slim)](https://docs.renovatebot.com/merge-confidence/) | [![adoption](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/adoption-slim)](https://docs.renovatebot.com/merge-confidence/) | [![passing](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/compatibility-slim/1.13.1)](https://docs.renovatebot.com/merge-confidence/) | [![confidence](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/confidence-slim/1.13.1)](https://docs.renovatebot.com/merge-confidence/) | | [torch](https://pytorch.org/) | `==1.12.1` -> `==2.0.0` | [![age](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/age-slim)](https://docs.renovatebot.com/merge-confidence/) | [![adoption](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/adoption-slim)](https://docs.renovatebot.com/merge-confidence/) | [![passing](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/compatibility-slim/1.12.1)](https://docs.renovatebot.com/merge-confidence/) | [![confidence](https://badges.renovateapi.com/packages/pypi/torch/2.0.0/confidence-slim/1.12.1)](https://docs.renovatebot.com/merge-confidence/) | --- ### ⚠ Dependency Lookup Warnings ⚠ Warnings were logged while processing this repo. Please check the Dependency Dashboard for more information. --- ### Configuration 📅 **Schedule**: Branch creation - At any time (no schedule defined), Automerge - At any time (no schedule defined). 🚦 **Automerge**: Disabled by config. Please merge this manually once you are satisfied. ♻ **Rebasing**: Never, or you tick the rebase/retry checkbox. 🔕 **Ignore**: Close this PR and you won't be reminded about these updates again. --- - [ ] If you want to rebase/retry this PR, check this box --- This PR has been generated by [Mend Renovate](https://www.mend.io/free-developer-tools/renovate/). View repository job log [here](https://app.renovatebot.com/dashboard#github/GoogleCloudPlatform/python-docs-samples).
diff --git a/dataflow/gpu-examples/pytorch-minimal/Dockerfile b/dataflow/gpu-examples/pytorch-minimal/Dockerfile
@@ -12,23 +12,20 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-FROM pytorch/pytorch:1.9.1-cuda11.1-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 WORKDIR /pipeline
 
 COPY requirements.txt .
 COPY *.py ./
 
-RUN apt-get update \
-    && apt-get install -y --no-install-recommends g++ \
-    && rm -rf /var/lib/apt/lists/* \
-    # Install the pipeline requirements and check that there are no conflicts.
-    # Since the image already has all the dependencies installed,
-    # there's no need to run with the --requirements_file option.
-    && pip install --no-cache-dir --upgrade pip \
+# Install the pipeline requirements and check that there are no conflicts.
+# Since the image already has all the dependencies installed,
+# there's no need to run with the --requirements_file option.
+RUN pip install --no-cache-dir --upgrade pip \
     && pip install --no-cache-dir -r requirements.txt \
     && pip check
 
 # Set the entrypoint to Apache Beam SDK worker launcher.
-COPY --from=apache/beam_python3.8_sdk:2.38.0 /opt/apache/beam /opt/apache/beam
+COPY --from=apache/beam_python3.10_sdk:2.48.0 /opt/apache/beam /opt/apache/beam
 ENTRYPOINT [ "/opt/apache/beam/boot" ]
diff --git a/dataflow/gpu-examples/pytorch-minimal/noxfile_config.py b/dataflow/gpu-examples/pytorch-minimal/noxfile_config.py
@@ -22,10 +22,10 @@
 
 TEST_CONFIG_OVERRIDE = {
     # You can opt out from the test for specific Python versions.
-    # > ℹ️ We're opting out of all Python versions except 3.8.
+    # > ℹ️ We're opting out of all Python versions except 3.10.
     # > The Python version used is defined by the Dockerfile, so it's redundant
     # > to run multiple tests since they would all be running the same Dockerfile.
-    "ignored_versions": ["2.7", "3.6", "3.7", "3.9", "3.10", "3.11"],
+    "ignored_versions": ["2.7", "3.6", "3.7", "3.8", "3.9", "3.11"],
     # Old samples are opted out of enforcing Python type hints
     # All new samples should feature them
     "enforce_type_hints": True,
diff --git a/dataflow/gpu-examples/pytorch-minimal/requirements.txt b/dataflow/gpu-examples/pytorch-minimal/requirements.txt
@@ -1,2 +1,2 @@
-apache-beam[gcp]==2.35.0
-torch==1.13.1
+apache-beam[gcp]==2.48.0
+triton==2.0.0
diff --git a/people-and-planet-ai/weather-forecasting/serving/requirements.txt b/people-and-planet-ai/weather-forecasting/serving/requirements.txt
@@ -1,6 +1,5 @@
 Flask==2.2.2
 gunicorn==20.1.0
-torch==1.13.1
 
 # Local packages.
 ./weather-data
diff --git a/people-and-planet-ai/weather-forecasting/tests/predictions_tests/requirements.txt b/people-and-planet-ai/weather-forecasting/tests/predictions_tests/requirements.txt
@@ -1,4 +1,3 @@
 ../../serving/weather-data
 ../../serving/weather-model
 plotly==5.15.0
-torch==1.13.1
