get torch 2.2.0 working

#3
by multimodalart HF staff - opened
Files changed (1) hide show
  1. 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.20.5
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.3.0
64
- torchvision==0.18.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