From 29f907060f87269915a4c236b948a1ac46cbb4d3 Mon Sep 17 00:00:00 2001 From: bxdd Date: Thu, 16 Nov 2023 20:54:14 +0800 Subject: [PATCH 1/3] [ENH] add CudaManager --- learnware/utils/gpu.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/learnware/utils/gpu.py b/learnware/utils/gpu.py index 95fbfe1..395d979 100644 --- a/learnware/utils/gpu.py +++ b/learnware/utils/gpu.py @@ -44,3 +44,17 @@ def choose_device(cuda_idx=-1): else: device = torch.device("cuda:0") return device + + +class CudaManager: + def __init__(self): + pass + + def get_specification_cuda_idx(self): + pass + + def get_cpu_idx(self): + pass + + def get_random_cuda_idx(self): + pass From ba9bc43ca60b6edf52a07c6b69bcc07e14f2a518 Mon Sep 17 00:00:00 2001 From: bxdd Date: Thu, 16 Nov 2023 22:03:48 +0800 Subject: [PATCH 2/3] [ENH] make setup install full package --- setup.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/setup.py b/setup.py index 705000d..8c6b36d 100644 --- a/setup.py +++ b/setup.py @@ -69,6 +69,7 @@ REQUIRED = [ "transformers>=4.34.1", "portalocker>=2.0.0", "qpsolvers[clarabel]>=4.0.1", + "geatpy>=2.7.0;python_version<='3.11'", ] here = os.path.abspath(os.path.dirname(__file__)) @@ -105,6 +106,14 @@ if __name__ == "__main__": "black==23.1.0", "pre-commit", ], + "full": [ + "torch>=1.11.0", + "torchvision>=0.15.1", + "torch-optimizer>=0.3.0", + "lightgbm>=3.3.0", + "torchvision>=0.15.1", + "sentence_transformers>=2.2.2", + ], }, classifiers=[ "Intended Audience :: Science/Research", From bb050b06d7a898714df0034f627570038beda346 Mon Sep 17 00:00:00 2001 From: bxdd Date: Thu, 16 Nov 2023 22:05:37 +0800 Subject: [PATCH 3/3] [MNT] update setup --- setup.py | 1 - 1 file changed, 1 deletion(-) diff --git a/setup.py b/setup.py index 8c6b36d..b6aea9a 100644 --- a/setup.py +++ b/setup.py @@ -111,7 +111,6 @@ if __name__ == "__main__": "torchvision>=0.15.1", "torch-optimizer>=0.3.0", "lightgbm>=3.3.0", - "torchvision>=0.15.1", "sentence_transformers>=2.2.2", ], },