RAPID/environment.yml at main · Costaki33/RAPID · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
name: rapid
channels:
  - defaults
dependencies:
  - python=3.10
  - pip
  - readline
  - sqlite
  - tk
  - zlib
  - xz
  - ncurses
  - ca-certificates
  - openssl
  - pip:
      # eqcctpro is vendored inside this repo (RAPID/eqcctpro); the PyPI
      # package of the same name is deprecated -- do not install it.
      - numpy==1.26.4
      - scipy==1.15.1
      - pandas==2.2.3
      - matplotlib==3.10.0
      - psutil==6.1.1
      - obspy==1.4.1
      - ray==2.42.1
      - plotly
      - seaborn
      # TensorFlow 2.15.1 + CUDA 12 stack
      - tensorflow==2.15.1
      - keras==2.15.0
      - tensorboard==2.15.2
      - tensorboard-data-server==0.7.2
      - protobuf>=4.25.0
      - h5py>=3.10.0
      # PyTorch + SeisBench stack
      - torch==2.5.1
      - torchvision==0.20.1
      - torchaudio==2.5.1
      - seisbench==0.10.2
      # CUDA 12 wheels (For TF 2.15.1 - requires cuDNN 8.9.x)
      - nvidia-cuda-runtime-cu12==12.1.105
      - nvidia-cublas-cu12==12.1.3.1
      - nvidia-cuda-nvrtc-cu12==12.1.105
      - nvidia-cufft-cu12==11.0.2.54
      - nvidia-curand-cu12==10.3.2.106
      - nvidia-cusolver-cu12==11.4.5.107
      - nvidia-cusparse-cu12==12.1.0.106
      - nvidia-cuda-cupti-cu12==12.1.105
      - nvidia-cudnn-cu12==8.9.7.29
      - nvidia-nccl-cu12==2.21.5
      - nvidia-nvjitlink-cu12==12.2.140
      - nvidia-nvtx-cu12==12.1.105
      - pynvml==12.0.0
      - shapely==2.1.0
      - requests==2.32.3
      - rich==13.9.4
      - silence-tensorflow==1.2.3