From d1493a3d1ed82e642a4baf142d91cbd41944cb99 Mon Sep 17 00:00:00 2001 From: Damien Mourot Date: Wed, 20 Mar 2024 19:42:15 -1000 Subject: [PATCH] feat(docker image): Optimized Docker image configuration for CPU-only environments --- backend/requirements.txt | 2 ++ 1 file changed, 2 insertions(+) diff --git a/backend/requirements.txt b/backend/requirements.txt index 91bf968d3..f6ff3eed0 100644 --- a/backend/requirements.txt +++ b/backend/requirements.txt @@ -1,4 +1,6 @@ -i https://pypi.org/simple +# Ensure PyTorch, along with its required dependencies, is explicitly installed in 'CPU-only' mode to optimize compatibility and performance for systems without GPU support. +--extra-index-url https://download.pytorch.org/whl/cpu aiohttp==3.9.3; python_version >= '3.8' aiosignal==1.3.1; python_version >= '3.7' amqp==5.2.0; python_version >= '3.6'