eqcctpro 0.3__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of eqcctpro might be problematic. Click here for more details.
- eqcctpro-0.3/PKG-INFO +25 -0
- eqcctpro-0.3/README.md +1 -0
- eqcctpro-0.3/eqcctpro/__init__.py +2 -0
- eqcctpro-0.3/eqcctpro/eqcctpro.py +64 -0
- eqcctpro-0.3/eqcctpro.egg-info/PKG-INFO +25 -0
- eqcctpro-0.3/eqcctpro.egg-info/SOURCES.txt +9 -0
- eqcctpro-0.3/eqcctpro.egg-info/dependency_links.txt +1 -0
- eqcctpro-0.3/eqcctpro.egg-info/requires.txt +21 -0
- eqcctpro-0.3/eqcctpro.egg-info/top_level.txt +1 -0
- eqcctpro-0.3/setup.cfg +4 -0
- eqcctpro-0.3/setup.py +38 -0
eqcctpro-0.3/PKG-INFO
ADDED
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
Metadata-Version: 2.2
|
|
2
|
+
Name: eqcctpro
|
|
3
|
+
Version: 0.3
|
|
4
|
+
Requires-Dist: numpy==1.26.4
|
|
5
|
+
Requires-Dist: pandas==2.2.3
|
|
6
|
+
Requires-Dist: matplotlib==3.10.0
|
|
7
|
+
Requires-Dist: obspy==1.4.1
|
|
8
|
+
Requires-Dist: progress==1.6
|
|
9
|
+
Requires-Dist: psutil==6.1.1
|
|
10
|
+
Requires-Dist: ray==2.42.1
|
|
11
|
+
Requires-Dist: schedule==1.2.2
|
|
12
|
+
Requires-Dist: sdnotify==0.3.2
|
|
13
|
+
Requires-Dist: tensorflow==2.15.1
|
|
14
|
+
Requires-Dist: tensorflow-estimator==2.15.0
|
|
15
|
+
Requires-Dist: tensorflow-io-gcs-filesystem==0.37.1
|
|
16
|
+
Requires-Dist: tensorboard==2.15.2
|
|
17
|
+
Requires-Dist: tensorboard-data-server==0.7.2
|
|
18
|
+
Requires-Dist: silence-tensorflow==1.2.3
|
|
19
|
+
Requires-Dist: scipy==1.15.1
|
|
20
|
+
Requires-Dist: protobuf==4.25.6
|
|
21
|
+
Requires-Dist: grpcio==1.70.0
|
|
22
|
+
Requires-Dist: absl-py==2.1.0
|
|
23
|
+
Requires-Dist: h5py==3.12.1
|
|
24
|
+
Requires-Dist: pynvml==12.0.0
|
|
25
|
+
Dynamic: requires-dist
|
eqcctpro-0.3/README.md
ADDED
|
@@ -0,0 +1 @@
|
|
|
1
|
+
hello!
|
|
@@ -0,0 +1,64 @@
|
|
|
1
|
+
|
|
2
|
+
import os
|
|
3
|
+
from predictor import EQCCTMSeedRunner, EvaluateSystem, OptimalCPUConfigurationFinder, OptimalGPUConfigurationFinder
|
|
4
|
+
input_mseed_directory_path = '/home/skevofilaxc/eqcctpro/mseed/20241215T115800Z_20241215T120100Z'
|
|
5
|
+
output_pick_directory_path = '/home/skevofilaxc/eqcctpro/outputs'
|
|
6
|
+
log_file_path = '/home/skevofilaxc/eqcctpro/outputs/eqcctpro.log'
|
|
7
|
+
csv_filepath = '/home/skevofilaxc/eqcctpro/csv'
|
|
8
|
+
|
|
9
|
+
# Can run EQCCT on a given input dir on GPU or CPU
|
|
10
|
+
# Can also specify the number of stations you want to use as well
|
|
11
|
+
|
|
12
|
+
# eqcct_runner = EQCCTMSeedRunner(use_gpu=True,
|
|
13
|
+
# intra_threads=1,
|
|
14
|
+
# inter_threads=1,
|
|
15
|
+
# cpu_id_list=[0,1,2,3,4],
|
|
16
|
+
# input_dir=input_mseed_directory_path,
|
|
17
|
+
# output_dir=output_pick_directory_path,
|
|
18
|
+
# log_filepath=log_file_path,
|
|
19
|
+
# P_threshold=0.001,
|
|
20
|
+
# S_threshold=0.02,
|
|
21
|
+
# p_model_filepath='/home/skevofilaxc/model/ModelPS/test_trainer_024.h5',
|
|
22
|
+
# s_model_filepath='/home/skevofilaxc/model/ModelPS/test_trainer_021.h5',
|
|
23
|
+
# number_of_concurrent_predictions=5,
|
|
24
|
+
# best_usecase_config=True,
|
|
25
|
+
# csv_dir=csv_filepath,
|
|
26
|
+
# selected_gpus=[0],
|
|
27
|
+
# set_vram_mb=24750,
|
|
28
|
+
# specific_stations='AT01, BP01, DG05')
|
|
29
|
+
|
|
30
|
+
# eqcct_runner.run_eqcctpro()
|
|
31
|
+
|
|
32
|
+
|
|
33
|
+
# eval_gpu = EvaluateSystem('gpu',
|
|
34
|
+
# intra_threads=1,
|
|
35
|
+
# inter_threads=1,
|
|
36
|
+
# input_dir=input_mseed_directory_path,
|
|
37
|
+
# output_dir=output_pick_directory_path,
|
|
38
|
+
# log_filepath=log_file_path,
|
|
39
|
+
# csv_dir=csv_filepath,
|
|
40
|
+
# P_threshold=0.001,
|
|
41
|
+
# S_threshold=0.02,
|
|
42
|
+
# p_model_filepath='/home/skevofilaxc/model/ModelPS/test_trainer_024.h5',
|
|
43
|
+
# s_model_filepath='/home/skevofilaxc/model/ModelPS/test_trainer_021.h5',
|
|
44
|
+
# stations2use=2,
|
|
45
|
+
# cpu_id_list=range(0,1,2),
|
|
46
|
+
# set_vram_mb=24750,
|
|
47
|
+
# selected_gpus=[0])
|
|
48
|
+
# eval_gpu.evaluate() # This triggers evaluate_gpu() if mode is 'gpu'
|
|
49
|
+
|
|
50
|
+
cpu_finder = OptimalCPUConfigurationFinder(csv_filepath)
|
|
51
|
+
best_cpu_config = cpu_finder.find_best_overall_usecase()
|
|
52
|
+
print(best_cpu_config)
|
|
53
|
+
|
|
54
|
+
|
|
55
|
+
optimal_cpu_config = cpu_finder.find_optimal_for(cpu=3, station_count=2)
|
|
56
|
+
print(optimal_cpu_config)
|
|
57
|
+
|
|
58
|
+
|
|
59
|
+
gpu_finder = OptimalGPUConfigurationFinder(csv_filepath)
|
|
60
|
+
best_gpu_config = gpu_finder.find_best_overall_usecase()
|
|
61
|
+
print(best_gpu_config)
|
|
62
|
+
|
|
63
|
+
optimal_gpu_config = gpu_finder.find_optimal_for(num_cpus=1, gpu_list=[0], station_count=1)
|
|
64
|
+
print(optimal_gpu_config)
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
Metadata-Version: 2.2
|
|
2
|
+
Name: eqcctpro
|
|
3
|
+
Version: 0.3
|
|
4
|
+
Requires-Dist: numpy==1.26.4
|
|
5
|
+
Requires-Dist: pandas==2.2.3
|
|
6
|
+
Requires-Dist: matplotlib==3.10.0
|
|
7
|
+
Requires-Dist: obspy==1.4.1
|
|
8
|
+
Requires-Dist: progress==1.6
|
|
9
|
+
Requires-Dist: psutil==6.1.1
|
|
10
|
+
Requires-Dist: ray==2.42.1
|
|
11
|
+
Requires-Dist: schedule==1.2.2
|
|
12
|
+
Requires-Dist: sdnotify==0.3.2
|
|
13
|
+
Requires-Dist: tensorflow==2.15.1
|
|
14
|
+
Requires-Dist: tensorflow-estimator==2.15.0
|
|
15
|
+
Requires-Dist: tensorflow-io-gcs-filesystem==0.37.1
|
|
16
|
+
Requires-Dist: tensorboard==2.15.2
|
|
17
|
+
Requires-Dist: tensorboard-data-server==0.7.2
|
|
18
|
+
Requires-Dist: silence-tensorflow==1.2.3
|
|
19
|
+
Requires-Dist: scipy==1.15.1
|
|
20
|
+
Requires-Dist: protobuf==4.25.6
|
|
21
|
+
Requires-Dist: grpcio==1.70.0
|
|
22
|
+
Requires-Dist: absl-py==2.1.0
|
|
23
|
+
Requires-Dist: h5py==3.12.1
|
|
24
|
+
Requires-Dist: pynvml==12.0.0
|
|
25
|
+
Dynamic: requires-dist
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
numpy==1.26.4
|
|
2
|
+
pandas==2.2.3
|
|
3
|
+
matplotlib==3.10.0
|
|
4
|
+
obspy==1.4.1
|
|
5
|
+
progress==1.6
|
|
6
|
+
psutil==6.1.1
|
|
7
|
+
ray==2.42.1
|
|
8
|
+
schedule==1.2.2
|
|
9
|
+
sdnotify==0.3.2
|
|
10
|
+
tensorflow==2.15.1
|
|
11
|
+
tensorflow-estimator==2.15.0
|
|
12
|
+
tensorflow-io-gcs-filesystem==0.37.1
|
|
13
|
+
tensorboard==2.15.2
|
|
14
|
+
tensorboard-data-server==0.7.2
|
|
15
|
+
silence-tensorflow==1.2.3
|
|
16
|
+
scipy==1.15.1
|
|
17
|
+
protobuf==4.25.6
|
|
18
|
+
grpcio==1.70.0
|
|
19
|
+
absl-py==2.1.0
|
|
20
|
+
h5py==3.12.1
|
|
21
|
+
pynvml==12.0.0
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
eqcctpro
|
eqcctpro-0.3/setup.cfg
ADDED
eqcctpro-0.3/setup.py
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
1
|
+
from setuptools import setup, find_packages
|
|
2
|
+
import os
|
|
3
|
+
|
|
4
|
+
# Ensure TensorFlow and CUDA are initialized before running
|
|
5
|
+
os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
|
|
6
|
+
os.environ["TF_ENABLE_ONEDNN_OPTS"] = "1"
|
|
7
|
+
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
|
|
8
|
+
os.environ["NVIDIA_LOG_LEVEL"] = "ERROR"
|
|
9
|
+
os.environ["CUDA_MODULE_LOADING"] = "LAZY"
|
|
10
|
+
|
|
11
|
+
setup(
|
|
12
|
+
name="eqcctpro",
|
|
13
|
+
version="0.3",
|
|
14
|
+
packages=find_packages(),
|
|
15
|
+
install_requires=[
|
|
16
|
+
"numpy==1.26.4",
|
|
17
|
+
"pandas==2.2.3",
|
|
18
|
+
"matplotlib==3.10.0",
|
|
19
|
+
"obspy==1.4.1",
|
|
20
|
+
"progress==1.6",
|
|
21
|
+
"psutil==6.1.1",
|
|
22
|
+
"ray==2.42.1",
|
|
23
|
+
"schedule==1.2.2",
|
|
24
|
+
"sdnotify==0.3.2",
|
|
25
|
+
"tensorflow==2.15.1",
|
|
26
|
+
"tensorflow-estimator==2.15.0",
|
|
27
|
+
"tensorflow-io-gcs-filesystem==0.37.1",
|
|
28
|
+
"tensorboard==2.15.2",
|
|
29
|
+
"tensorboard-data-server==0.7.2",
|
|
30
|
+
"silence-tensorflow==1.2.3",
|
|
31
|
+
"scipy==1.15.1",
|
|
32
|
+
"protobuf==4.25.6",
|
|
33
|
+
"grpcio==1.70.0",
|
|
34
|
+
"absl-py==2.1.0",
|
|
35
|
+
"h5py==3.12.1",
|
|
36
|
+
"pynvml==12.0.0",
|
|
37
|
+
],
|
|
38
|
+
)
|