environment.yml

name: behavior-driven-video-synthesis
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - absl-py=0.8.1=py37_0
  - conda=4.8.0=py37_1
  - cudatoolkit=10.1.243=h6bb024c_0
  - ffmpeg=4.1.3=h167e202_0
  - h5py=2.10.0=nompi_py37h513d04c_101
  - hdf5=1.10.5=nompi_h3c11f04_1104
  - ignite=0.3.0
  - imageio=2.6.1=py37_0
  - imagesize=1.1.0=py_0
  - ipython=7.10.2=py37h39e3cac_0
  - ipython_genutils=0.2.0=py_1
  - keras-applications=1.0.8=py_1
  - keras-preprocessing=1.1.0=py_0
  - libopencv=4.1.2=py37_2
  - numpy=1.17.4=py37hc1035e2_0
  - opencv=4.1.2=py37_2
  - opt_einsum=3.1.0=py_0
  - pandas=0.25.3=py37hb3f55d8_0
  - pillow=6.2.1=py37h34e0f95_0
  - pip=21.0.1
  - py-opencv=4.1.2=py37h5ca1d4c_2
  - pyqt=5.12.3=py37hcca6a23_1
  - python=3.7.5=h0371630_0
  - pytorch=1.3.1=py3.7_cuda10.1.243_cudnn7.6.3_0
  - pyyaml=5.1.2=py37h516909a_0
  - qt=5.12.5=hd8c4c69_1
  - scikit-image=0.16.2=py37hb3f55d8_0
  - scikit-learn=0.22=py37hcdab131_1
  - scikit-video=1.1.11=pyh24bf2e0_0
  - scipy=1.3.3
  - tensorboard=2.0.0=pyhb38c66f_1
  - tensorflow=2.0.0=mkl_py37h66b46cc_0
  - tensorflow-base=2.0.0=mkl_py37h9204916_0
  - tensorflow-estimator=2.0.0=pyh2649769_0
  - torchvision
  - tqdm=4.40.2=py_0
  - yaml=0.1.7=had09818_2
  - pip:
    - gpustat==0.6.0
    - imageio-ffmpeg==0.3.0
    - kornia==0.2.0
    - matplotlib==3.2.1
    - moviepy==1.0.1
    - pyqt5-sip==4.19.18
    - umap-learn==0.3.10
    - wandb==0.8.31
    - cdflib