environment.yml

name: gmd
channels:
  - defaults
  - pytorch
  - conda-forge
  - nvidia
dependencies:
  - python=3.9.6=h12debd9_1
  - cudatoolkit=11.1.74=h6bb024c_0
  - gpustat=0.6.0=pyhd3eb1b0_1
  - imageio=2.9.0=pyhd3eb1b0_0
  - pillow=8.3.1=py39h5aabda8_0
  - pip=21.2.4=py37h06a4308_0
  - pytorch=1.10.1=py3.9_cuda11.1_cudnn8.0.5_0
  - pytorch-mutex=1.0=cuda
  - pywavelets=1.1.1=py39h6323ea4_4
  - scikit-image=0.16.2=py39ha9443f7_0
  - torchaudio=0.10.1=py39_cu111
  - torchvision=0.11.2=py39_cu111
  - tornado=6.1=py39h27cfd23_0
  - pip:
    - einops==0.3.2
    - gradio==3.13.0
    - h5py==3.7.0
    - hydra-colorlog==1.1.0
    - hydra-core==1.2.0
    - ipython==7.27.0
    - kornia==0.5.8
    - matplotlib==3.4.3
    - medpy==0.4.0
    - monai==0.7.0
    - nibabel==3.2.1
    - numpy==1.20.3
    - opencv-python==4.5.4.60
    - openmim==0.3.6
    - ordered-set==4.1.0
    - pandas==1.3.2
    - pytorch-lightning==1.5.7
    - regex==2022.8.17
    - scikit-learn==1.0.2
    - scipy==1.7.1
    - seaborn==0.11.2
    - simpleitk==2.1.0
    - tensorboard==2.6.0
    - tensorboardX==2.6.2.2
    - torchmetrics==0.6.0
    - tqdm==4.62.1