Spaces:
Running
on
Zero
Running
on
Zero
get torch 2.2.0 working
#3
by
multimodalart
HF staff
- opened
- requirements.txt +3 -3
requirements.txt
CHANGED
@@ -38,7 +38,7 @@ nvidia-cufft-cu12==11.0.2.54
|
|
38 |
nvidia-curand-cu12==10.3.2.106
|
39 |
nvidia-cusolver-cu12==11.4.5.107
|
40 |
nvidia-cusparse-cu12==12.1.0.106
|
41 |
-
nvidia-nccl-cu12==2.
|
42 |
nvidia-nvjitlink-cu12==12.5.40
|
43 |
nvidia-nvtx-cu12==12.1.105
|
44 |
pandas==2.2.2
|
@@ -60,8 +60,8 @@ setproctitle==1.3.3
|
|
60 |
smmap==5.0.1
|
61 |
sympy==1.12.1
|
62 |
tokenizers==0.19.1
|
63 |
-
torch==2.
|
64 |
-
torchvision==0.
|
65 |
tqdm==4.66.4
|
66 |
transformers==4.41.1
|
67 |
typing-extensions==4.12.0
|
|
|
38 |
nvidia-curand-cu12==10.3.2.106
|
39 |
nvidia-cusolver-cu12==11.4.5.107
|
40 |
nvidia-cusparse-cu12==12.1.0.106
|
41 |
+
nvidia-nccl-cu12==2.19.3
|
42 |
nvidia-nvjitlink-cu12==12.5.40
|
43 |
nvidia-nvtx-cu12==12.1.105
|
44 |
pandas==2.2.2
|
|
|
60 |
smmap==5.0.1
|
61 |
sympy==1.12.1
|
62 |
tokenizers==0.19.1
|
63 |
+
torch==2.2.0
|
64 |
+
torchvision==0.17.0
|
65 |
tqdm==4.66.4
|
66 |
transformers==4.41.1
|
67 |
typing-extensions==4.12.0
|