File size: 866 Bytes
28451f7
 
cbe7167
28451f7
5fa8a70
28451f7
 
 
 
 
 
 
cbe7167
 
 
 
 
cf41009
f52a4d0
 
cbe7167
8bdfbb7
5fa8a70
701e903
5fa8a70
 
d708e6e
3cc0f0e
7104534
579d922
33eb31d
61d27a8
cbe7167
8bf5f16
17c6444
cf41009
cbe7167
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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