YinuoGuo27 commited on
Commit
4743c75
·
verified ·
1 Parent(s): 7beb039

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -7
requirements.txt CHANGED
@@ -1,11 +1,9 @@
1
- # python=3.10
2
- # conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia
3
- torch==2.2.2
4
- torchvision==0.17.2
5
- torchaudio==2.2.2
6
- onnxruntime-gpu==1.18.0
7
  tensorrt
8
- transformers==4.33.0
9
  torchgeometry
10
  pyyaml==6.0.1
11
  scipy==1.11.4
 
1
+ torch
2
+ torchvision
3
+ torchaudio
4
+ onnxruntime-gpu
 
 
5
  tensorrt
6
+ transformers
7
  torchgeometry
8
  pyyaml==6.0.1
9
  scipy==1.11.4