environment_docker.yml

name: masaenv
channels:
  - conda-forge
  - pytorch
  - nvidia
  - defaults
dependencies:
  - python=3.11
  - pytorch=2.1.2
  - torchvision=0.16.2
  - torchaudio=2.1.2
  - pytorch-cuda=11.8
  - cudatoolkit=11.8
  - ffmpeg
  - pip
  - pip:
      - fsspec
      - gradio
      - opencv-python