search

AI4EPS / PhaseNet

PhaseNet: A Deep-Neural-Network-Based Seismic Arrival Time Picking Method
MIT License
268 stars 111 forks source link

PhaseNet on DGX A100 GPU #72

Open fizzi opened 1 year ago

fizzi commented 1 year ago

Hello everyone,

Is there a way to run PhaseNet on gpu?

I'd like to run it on NVIDIA DGX A100 80gb.

Thanks,

zhuwq0 commented 1 year ago

Hi, I think you can follow this instruction to install the gpu version of tensorflow: https://www.tensorflow.org/install/pip