Requirements / conda environment file?

[amonod]

Hi,

Is there any chance you could share all required packages to run everything contained in the repo (code + jupyter notebooks), either through a simple dotted list, or with a .yml conda environment file (e.g. as done here: https://github.com/m-tassano/fastdvdnet/blob/master/requirements.yml)?

Thank you and congratulations for your very interesting publication, Antoine

[amonod]

If that may help, here are the contents of a conda environment file udvd_conda_env.yml I created based on the imports found in the .py and .ipynb found in the repo. I haven't tried everything so some packages could be missing or using the wrong version. Feel free to add it yourself or through a pull request if needed.

name: udvd
channels:
  - pytorch-lts
  - nvidia
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - cudatoolkit=11.1
  - h5py
  - jupyterlab=3.1.1
  - matplotlib
  - numpy
  - pandas
  - pip
  - py-opencv=3.4.2
  - python
  - pytorch
  - scikit-image
  - seaborn
  - tensorboard
  - torchaudio
  - torchvision
  - tqdm