Li Zhaoxu commited on
Commit
c34b4fa
1 Parent(s): c295b91
Files changed (1) hide show
  1. requirements.txt +11 -1
requirements.txt CHANGED
@@ -69,7 +69,17 @@ nest-asyncio==1.5.8
69
  networkx==3.2.1
70
  nltk==3.8.1
71
  numpy==1.26.2
72
-
 
 
 
 
 
 
 
 
 
 
73
  omegaconf==2.3.0
74
  opencv-python==4.7.0.72
75
  orjson==3.9.10
 
69
  networkx==3.2.1
70
  nltk==3.8.1
71
  numpy==1.26.2
72
+ nvidia-cublas-cu11==11.10.3.66
73
+ nvidia-cuda-cupti-cu11==11.7.101
74
+ nvidia-cuda-nvrtc-cu11==11.7.99
75
+ nvidia-cuda-runtime-cu11==11.7.99
76
+ nvidia-cudnn-cu11==8.5.0.96
77
+ nvidia-cufft-cu11==10.9.0.58
78
+ nvidia-curand-cu11==10.2.10.91
79
+ nvidia-cusolver-cu11==11.4.0.1
80
+ nvidia-cusparse-cu11==11.7.4.91
81
+ nvidia-nccl-cu11==2.14.3
82
+ nvidia-nvtx-cu11==11.7.91
83
  omegaconf==2.3.0
84
  opencv-python==4.7.0.72
85
  orjson==3.9.10