deps:refactor dependencies installation method (#2381)

This commit is contained in:
yyhhyy
2025-03-02 18:34:01 +08:00
committed by GitHub
parent 904e8e0393
commit 0c7dae71da
7 changed files with 1906 additions and 694 deletions

View File

@@ -42,7 +42,6 @@ auto = [
"torchvision>=0.17.1; sys_platform != 'darwin' or platform_machine != 'x86_64'",
"torchvision>=0.17.1,<0.18.0; sys_platform == 'darwin' and platform_machine == 'x86_64'",
]
cpu = [
# Starting from PyTorch 2.3.0, the minimum requirement for macOS is macOS 11.0+ ARM64.
"torch>=2.2.1; sys_platform != 'darwin' or platform_machine != 'x86_64'",
@@ -52,7 +51,6 @@ cpu = [
"torchvision>=0.17.1; sys_platform != 'darwin' or platform_machine != 'x86_64'",
"torchvision>=0.17.1,<0.18.0; sys_platform == 'darwin' and platform_machine == 'x86_64'",
]
cuda118 = [
"torch>=2.2.1; sys_platform == 'win32' or sys_platform == 'linux'",
"torchvision>=0.17.1; sys_platform == 'win32' or sys_platform == 'linux'",
@@ -69,7 +67,6 @@ cuda124 = [
"torchvision>=0.17.1; sys_platform == 'win32' or sys_platform == 'linux'",
"torchaudio>=2.2.1; sys_platform == 'win32' or sys_platform == 'linux'",
]
#rocm60 = [
# "torch>=2.3.0,<2.4.0;sys_platform == 'linux'",
# "torchvision>=0.18.0,<0.19.0;sys_platform == 'linux'",
@@ -77,7 +74,7 @@ cuda124 = [
#]
vllm = [
# Just support GPU version on Linux
"vllm>=0.7.0; sys_platform == 'linux'"
"vllm>=0.7.0; sys_platform == 'linux'",
]
#vllm_pascal = [
# # https://github.com/sasha0552/pascal-pkgs-ci
@@ -178,7 +175,6 @@ torch = [
{ index = "pytorch-cu124", marker = "platform_system == 'Linux'", extra = "cuda124" },
# { index = "pytorch-rocm60", marker = "platform_system == 'Linux'", extra = "rocm60" },
]
torchvision = [
{ index = "pytorch-cpu", marker = "platform_system == 'Darwin'" },
{ index = "pytorch-cpu", marker = "platform_system == 'Windows'", extra = "cpu" },