name: cosmos-predict1
channels:
  # Keep these channels for other dependencies, but PyTorch will be installed via pip
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pip=25.0
  - cmake
  - ninja
  - gcc=12.4.0
  - gxx=12.4.0
  # Remove all PyTorch/TorchVision/CUDA related entries from Conda dependencies
  # - pytorch=2.3.1
  # - torchvision=0.18.1
  # - torchaudio=2.3.1
  # - pytorch-cuda=12.1
  - cudnn
  - cuda-toolkit=12.1
  - cuda-nvcc=12.1
  # - libcublas

  - pip:
      - -r file:///app/gui/requirements.txt
      - uvicorn[standard]
      - fastapi
      - git+https://github.com/microsoft/MoGe.git
      - einops
      - omegaconf
      - diffusers
      - megatron
      - IPython
      - megatron.core
      - attrs
      - iopath
      #- transformer_engine
      # PyTorch and TorchVision will be installed separately in the Dockerfile