From e55bdd7aee13b7423b61c34667d68ac431ce82de Mon Sep 17 00:00:00 2001 From: Jason Kneen Date: Tue, 13 Aug 2024 13:21:52 +0100 Subject: [PATCH] Enable GPU on tensorflow --- requirements.txt | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 9396c93..d459449 100644 --- a/requirements.txt +++ b/requirements.txt @@ -7,7 +7,8 @@ opencv-python==4.8.1.78 pillow==9.5.0 insightface==0.7.3 torch==2.1.0 # Add the specific version you're using -tensorflow==2.16.1 # Add the specific version you're using +tensorflow-macos==2.16.2 # Add the specific version you're using +tensorflow-metal==1.1.0 # Add the specific version you're using # Image processing scikit-image==0.24.0