Update requirements.txt
updated from CUDA 11.8 to CUDA 12.1
This commit is contained in:
committed by
GitHub
parent
db594c0e7c
commit
e862ff1456
@@ -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
|
numpy>=1.23.5,<2
|
||||||
opencv-python==4.10.0.84
|
opencv-python==4.10.0.84
|
||||||
@@ -11,7 +11,7 @@ customtkinter==5.2.2
|
|||||||
pillow==11.1.0
|
pillow==11.1.0
|
||||||
torch==2.0.1+cu118; sys_platform != 'darwin'
|
torch==2.0.1+cu118; sys_platform != 'darwin'
|
||||||
torch==2.0.1; 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'
|
torchvision==0.15.2; sys_platform == 'darwin'
|
||||||
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
|
onnxruntime-silicon==1.16.3; sys_platform == 'darwin' and platform_machine == 'arm64'
|
||||||
onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
|
onnxruntime-gpu==1.16.3; sys_platform != 'darwin'
|
||||||
|
|||||||
Reference in New Issue
Block a user