Use CPU versions of pyTorch

SkyTruth · Dec 24, 2024 · 93349d6 · 93349d6
1 parent 3fa8ed6
commit 93349d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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