Files changed (1) hide show
  1. requirements.txt +13 -2
requirements.txt CHANGED
@@ -1,3 +1,9 @@
 
 
 
 
 
 
1
  torch==2.1.0
2
  torchvision==0.16.0
3
  torchaudio==2.1.0
@@ -6,6 +12,7 @@ einops
6
  omegaconf
7
  deepspeed
8
  torchmetrics
 
9
  webdataset
10
  accelerate
11
  tensorboard
@@ -19,5 +26,9 @@ imageio[ffmpeg]
19
  xatlas
20
  plyfile
21
  xformers==0.0.22.post7
22
- git+https://github.com/NVlabs/nvdiffrast/
23
- huggingface-hub
 
 
 
 
 
1
+ # Build Tools
2
+ setuptools
3
+ wheel
4
+ ninja
5
+
6
+ # Core AI & 3D
7
  torch==2.1.0
8
  torchvision==0.16.0
9
  torchaudio==2.1.0
 
12
  omegaconf
13
  deepspeed
14
  torchmetrics
15
+ onnxruntime
16
  webdataset
17
  accelerate
18
  tensorboard
 
26
  xatlas
27
  plyfile
28
  xformers==0.0.22.post7
29
+
30
+ # Nvdiffrast (Compiles during install)
31
+ git+https://github.com
32
+
33
+ # Utility
34
+ huggingface-hub