cosmos-predict1.yaml

name: cosmos-predict1
channels:
  - pytorch          
  - nvidia            
  - conda-forge
  - defaults
dependencies:
  - python=3.10
  - pip=25.0
  - cmake
  - ninja
  - gcc=12.4.0
  - gxx=12.4.0
# REMOVE these explicit CUDA dependencies from here.
  # Let pytorch-cuda manage its own CUDA libraries.
  # - cuda=12.4
  # - cuda-nvcc=12.4
  # - cuda-toolkit=12.4
  # Add PyTorch and Torchvision
  - pytorch=2.3.1
  - torchvision=0.18.1
  - torchaudio=2.3.1
  - pytorch-cuda=12.1 

  # Add pip dependencies directly in the YAML
  - pip:
      - -r file:///app/gui/requirements.txt 
      - uvicorn[standard]
      - fastapi
      # Add any other crucial top-level pip dependencies here if your project needs them
      # Example: - accelerate
      # Example: - transformers