jiawei011 commited on
Commit
89d4d7f
1 Parent(s): a985a08

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -1
requirements.txt CHANGED
@@ -11,7 +11,9 @@ imageio
11
  imageio-ffmpeg
12
  omegaconf
13
 
14
- torch
 
 
15
  einops
16
  plyfile
17
  pygltflib
 
11
  imageio-ffmpeg
12
  omegaconf
13
 
14
+ torch==2.0.0 --index-url https://download.pytorch.org/whl/cu118
15
+ torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cu118
16
+ torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118
17
  einops
18
  plyfile
19
  pygltflib