SVGDTutorial/environment_gpu.yml at main · miguelcorralesg/SVGDTutorial · 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
name: svgdtutorial_gpu
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - python>=3.8
  - pip
  - numpy>=1.15.0
  - scipy>=1.4.0
  - cudatoolkit=11.8  # Update to the most recent CUDA version compatible with PyTorch
  - pytorch>=2.0  # Use the latest version of PyTorch compatible with CUDA 11.8
  - torchvision>=0.15  # Match the corresponding torchvision version
  - torchaudio>=2.0  # Match the corresponding torchaudio version
  - matplotlib
  - ipython
  - jupyter
  - pylops
  - tqdm
  - gstools
  - pip:
      - cupy-cuda11x  # Update to use a generic cupy package compatible with CUDA 11.x
      - setuptools_scm
      - torchsummary
      - segyio
      - h5py
      - juliacall
      - hydra-core
      - imageio
      - imageio[ffmpeg]
      - scikit-learn
      - seaborn
      - opencv-python
      - corner
      - hdbscan
      - tabulate
      - scikit-image
      - torchmetrics
      - RISE
      - nbconvert