updated extensions requirements

P2Enjoy · Nov 13, 2023 · 24e30bb · 24e30bb
1 parent f8a4484
commit 24e30bb
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 11 deletions.
diff --git a/data/extension-requirements.txt b/data/extension-requirements.txt
@@ -47,9 +47,18 @@ mediapipe==0.10.7
 moviepy==1.0.2
 networkx==3.2
 numba==0.58.1
-nvidia-cublas-cu12==12.3.2.9
-nvidia-cuda-nvrtc-cu12==12.3.52
-nvidia-cuda-runtime-cu12==12.3.52
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.52
+nvidia-nvtx-cu12==12.1.105
 onnx==1.14.0
 onnxruntime==1.15.0
 opencv-contrib-python==4.8.1.78
@@ -91,13 +100,17 @@ tb-nightly==2.16.0a20231024
 tensorboard-data-server==0.7.2
 threadpoolctl==3.2.0
 tinycss2==1.2.1
+torch==2.1.0
 torchmetrics==0.11.4
+torchvision==0.16.0
 traitlets==5.12.0
 transforms3d==0.4.1
 transparent-background==1.2.9
 trimesh==4.0.0
+triton==2.1.0
 urllib3==2.0.7
 vispy==0.13.0
 webencodings==0.5.1
 Werkzeug==3.0.1
+xformers==0.0.22.post7
 yacs==0.1.8
diff --git a/data/requirements.txt b/data/requirements.txt
@@ -125,11 +125,19 @@ ninja==1.11.1.1
 numba==0.58.1
 numpy==1.23.5
 nvidia-cublas-cu11==11.11.3.6
-nvidia-cublas-cu12==12.3.2.9
-nvidia-cuda-nvrtc-cu12==12.3.52
-nvidia-cuda-runtime-cu12==12.3.52
+nvidia-cublas-cu12==12.1.3.1
+nvidia-cuda-cupti-cu12==12.1.105
+nvidia-cuda-nvrtc-cu12==12.1.105
+nvidia-cuda-runtime-cu12==12.1.105
 nvidia-cudnn-cu11==8.6.0.163
-nvidia-cudnn-cu12==8.9.4.25
+nvidia-cudnn-cu12==8.9.2.26
+nvidia-cufft-cu12==11.0.2.54
+nvidia-curand-cu12==10.3.2.106
+nvidia-cusolver-cu12==11.4.5.107
+nvidia-cusparse-cu12==12.1.0.106
+nvidia-nccl-cu12==2.18.1
+nvidia-nvjitlink-cu12==12.3.52
+nvidia-nvtx-cu12==12.1.105
 nvidia-tensorrt==99.0.0
 oauthlib==3.2.2
 omegaconf==2.2.3
@@ -230,20 +238,20 @@ tokenizers==0.13.3
 tomesd==0.1.3
 tomli==2.0.1
 toolz==0.12.0
-torch==2.0.1+cu118
+torch==2.1.0
 torchaudio==2.0.2+cu118
 torchdiffeq==0.2.3
 torchmetrics==0.11.4
 torchsde==0.2.5
-torchvision==0.15.2+cu118
+torchvision==0.16.0
 tqdm==4.66.1
 traitlets==5.12.0
 trampoline==0.1.2
 transformers==4.30.2
 transforms3d==0.4.1
 transparent-background==1.2.9
 trimesh==4.0.0
-triton==2.0.0
+triton==2.1.0
 typing-inspect==0.9.0
 typing_extensions==4.5.0
 tzdata==2023.3
@@ -256,7 +264,7 @@ webencodings==0.5.1
 websockets==11.0.3
 Werkzeug==3.0.1
 wrapt==1.14.1
-xformers==0.0.20
+xformers==0.0.22.post7
 yacs==0.1.8
 yapf==0.40.2
 yarl==1.9.2