Update requirements.txt

updated from CUDA 11.8 to CUDA 12.1
pull/911/head
Kenneth Estanislao 2025-02-01 12:21:55 +08:00 committed by GitHub
parent db594c0e7c
commit e862ff1456
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions

View File

@ -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'