whaohan commited on
Commit
d077e81
1 Parent(s): 160cf28

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  meshgpt_pytorch==0.6.7
2
  pytorch-custom-utils==0.0.21
3
  accelerate>=0.25.0
@@ -20,7 +23,6 @@ pyrender
20
  open3d-python
21
  easydict
22
  chardet
23
- deepspeed
24
  omegaconf
25
  scikit-image
26
  setuptools
 
1
+ --find-links https://download.pytorch.org/whl/cu118
2
+ torch==2.1.2
3
+ torchvision==0.16.2
4
  meshgpt_pytorch==0.6.7
5
  pytorch-custom-utils==0.0.21
6
  accelerate>=0.25.0
 
23
  open3d-python
24
  easydict
25
  chardet
 
26
  omegaconf
27
  scikit-image
28
  setuptools