RishubhPar commited on
Commit
4bf6e25
·
verified ·
1 Parent(s): bd90279

updated requirements

Browse files
Files changed (1) hide show
  1. requirements.txt +12 -2
requirements.txt CHANGED
@@ -1,5 +1,15 @@
1
- --extra-index-url https://download.pytorch.org/whl/cu113
2
- torch
 
 
 
 
 
 
 
 
 
 
3
  absl-py==2.3.1
4
  accelerate==1.9.0
5
  annotated-types==0.7.0
 
1
+ nvidia-cublas-cu12==12.1.3.1
2
+ nvidia-cuda-cupti-cu12==12.1.105
3
+ nvidia-cuda-nvrtc-cu12==12.1.105
4
+ nvidia-cuda-runtime-cu12==12.1.105
5
+ nvidia-cudnn-cu12==8.9.2.26
6
+ nvidia-cufft-cu12==11.0.2.54
7
+ nvidia-curand-cu12==10.3.2.106
8
+ nvidia-cusolver-cu12==11.4.5.107
9
+ nvidia-cusparse-cu12==12.1.0.106
10
+ nvidia-nccl-cu12==2.19.3
11
+ nvidia-nvjitlink-cu12==12.5.40
12
+ nvidia-nvtx-cu12==12.1.105
13
  absl-py==2.3.1
14
  accelerate==1.9.0
15
  annotated-types==0.7.0