EQTransformer icon indicating copy to clipboard operation
EQTransformer copied to clipboard

GPU usage at the EQTransformer

Open birsenc opened this issue 1 year ago • 1 comments

Dear @smousavi05 and all,

We are running EQTransformer at HPC using CPU. We have more than 100 seismic stations and 3 years of data. We want to accelerate the speed of predicition. I have questions related to this issue.

Thank you very much in advance

  • Is it faster to use both CPU and GPU or only GPU?

  • How do we have to choose the below parameters for faster predicition?

  • What does "int" mean? What value do we have to use for the GPU?

  • We choose number of CPUs at the slurm. Do we still have to enter the number_of_cpus, if not does it use "default=5"?

  • I couldn't understand "Id of GPU used for the prediction"

  • Do we have to choose the below parameters and use exactly the same values at the slurm also?

    gpuid: int, default=None Id of GPU used for the prediction. If using CPU set to None.

    gpu_limit: int, default=None Set the maximum percentage of memory usage for the GPU.

    number_of_cpus: int, default=5 Number of CPUs used for the parallel preprocessing and feeding of data for prediction.

    use_multiprocessing: bool, default=True If True, multiple CPUs will be used for the preprocessing of data even when GPU is used for the prediction.

birsenc avatar Sep 20 '24 18:09 birsenc