name: nex 
dependencies:
  - python=3.8
  - pip
  - cudatoolkit=10.2
  - ffmpeg=4.2.2
  - numpy=1.18.1
  - pandas=1.0.3
  - scipy=1.4.1
  - scikit-image=0.16.2
  - imageio==2.9.0
  - pip:  
    - opencv-python==4.1.2.30
    - lpips==0.1.3
    - tqdm==4.59.0 #required by LPIPS
    - tensorboard==2.4.1
    # conda give error about cuda. So, we install torch via pip instead. 
    - torch==1.8.0
    - torchvision==0.9.0