From e862ff1456be5cdbf2390f0e1caae307194acd00 Mon Sep 17 00:00:00 2001 From: Kenneth Estanislao Date: Sat, 1 Feb 2025 12:21:55 +0800 Subject: [PATCH] Update requirements.txt updated from CUDA 11.8 to CUDA 12.1 --- requirements.txt | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 5c04a4d..635b5a1 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,4 @@ ---extra-index-url https://download.pytorch.org/whl/cu118 +--extra-index-url https://download.pytorch.org/whl/cu121 numpy>=1.23.5,<2 opencv-python==4.10.0.84 @@ -11,7 +11,7 @@ customtkinter==5.2.2 pillow==11.1.0 torch==2.0.1+cu118; sys_platform != 'darwin' torch==2.0.1; sys_platform == 'darwin' -torchvision==0.15.2+cu118; sys_platform != 'darwin' +torchvision==0.15.2+cu121; sys_platform != 'darwin' torchvision==0.15.2; sys_platform == 'darwin' onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64' onnxruntime-gpu==1.16.3; sys_platform != 'darwin'