environment.yml

# Create environment with `conda env create -f environment.yml`
# Activate environment with `conda activate vgiwae`
# Update environment with `conda env update --file environment.yml`
# Remove environment with `conda remove --name vgiwae --all`

name: vgiwae
channels:
    - pytorch
    - conda-forge
    - anaconda
dependencies:
    - python=3.8
    - flake8
    - pytorch=1.10.1
    - torchvision=0.11.2
    # To address a bug in PyTorch 1.10.1 https://stackoverflow.com/questions/70520120/attributeerror-module-setuptools-distutils-has-no-attribute-version
    # downgrading the setuptools to below version
    - setuptools=59.5.0
    # - torchquad=0.2.3
    - pip
    - pip:
        - pytorch-lightning==1.5.8
        # - pytorch-lightning==1.6.5
        - jsonargparse[signatures]==4.1.2
        # - jsonargparse[signatures]==4.7.1
        - einops==0.3.2
    - tqdm
    - numpy=1.21.2
    - scipy=1.7.3
    - ipython
    - matplotlib
    - seaborn
    - ipykernel
    - ipywidgets