environment.yml

---
name: armory-core
channels:
  - pytorch
  - conda-forge
  - defaults
dependencies:
  - pytorch=1.12
  - torchvision
  - torchaudio
  - cudatoolkit=11.6
  - scikit-learn=1.0  # ART requires scikit-learn >=0.22.2,<1.1.0
  - jupyterlab
  - jupyterlab_widgets
  - ipywidgets
  - matplotlib
  - conda-forge::ffmpeg  # conda-forge ffmpeg comes with libx264 encoder, which the pytorch channel version does not include. This encoder is required for video compression defenses (ART) and video exporting. Future work could migrate this to libopenh264 encoder, which is available in both channels.
  - librosa
  - cudnn  # cudnn required for tensorflow
  - pandas
  - protobuf
prefix: /opt/conda