From 93349d6dfdb3ba515ff569fd7fb0ca5ae4ce6a59 Mon Sep 17 00:00:00 2001 From: Anthony Lukach Date: Mon, 23 Dec 2024 20:23:01 -0800 Subject: [PATCH] Use CPU versions of pyTorch --- cerulean_cloud/cloud_run_offset_tiles/requirements.txt | 2 +- cerulean_cloud/cloud_run_orchestrator/requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cerulean_cloud/cloud_run_offset_tiles/requirements.txt b/cerulean_cloud/cloud_run_offset_tiles/requirements.txt index 41d66c0c..2ba4b9f1 100644 --- a/cerulean_cloud/cloud_run_offset_tiles/requirements.txt +++ b/cerulean_cloud/cloud_run_offset_tiles/requirements.txt @@ -3,7 +3,7 @@ starlette-cramjam==0.1.0 uvicorn[standard] rasterio==1.3.0 setuptools==59.5.0 -torch==1.11.0 +torch==1.11.0+cpu torchvision==0.12.0 numpy<2.0.0 typing-inspect diff --git a/cerulean_cloud/cloud_run_orchestrator/requirements.txt b/cerulean_cloud/cloud_run_orchestrator/requirements.txt index 4a220b2e..2adc6251 100644 --- a/cerulean_cloud/cloud_run_orchestrator/requirements.txt +++ b/cerulean_cloud/cloud_run_orchestrator/requirements.txt @@ -24,7 +24,7 @@ pygeos networkx google-cloud-tasks protobuf -torch==1.11.0 +torch==1.11.0+cpu torchvision==0.12.0 scipy scikit-image