lucas-ventura commited on
Commit
b571e1d
·
verified ·
1 Parent(s): 2102cbc

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -1,5 +1,7 @@
1
  torch==2.5.1
2
  torchaudio==2.5.1
 
 
3
  torchmetrics
4
  lightning
5
  hydra-core==1.3.2
 
1
  torch==2.5.1
2
  torchaudio==2.5.1
3
+ nvidia-pyindex
4
+ nvidia-cudnn
5
  torchmetrics
6
  lightning
7
  hydra-core==1.3.2