Spaces:
Build error
Build error
| 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 |