YinuoGuo27 commited on
Commit
7bfd5cc
·
verified ·
1 Parent(s): 339fb2c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +1 -1
requirements.txt CHANGED
@@ -2,7 +2,7 @@ torch
2
  torchvision
3
  torchaudio
4
  onnx
5
- onnxruntime-gpu
6
  tensorrt
7
  transformers
8
  torchgeometry
 
2
  torchvision
3
  torchaudio
4
  onnx
5
+ onnxruntime-gpu==1.17.0
6
  tensorrt
7
  transformers
8
  torchgeometry