BoxOfColors commited on
Commit
5d5b191
·
1 Parent(s): 3f83c4f

Pin torch to exact cu128 build to match torchaudio

Browse files

Without an explicit torch pin, pip resolved a torch build that didn't
ship libcudart.so.12, so torchaudio's compiled CUDA extension failed
to load at runtime (RUNTIME_ERROR, not build error).

Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -21,5 +21,6 @@ torchdiffeq
21
  open_clip_torch
22
  git+https://github.com/descriptinc/audiotools
23
  --extra-index-url https://download.pytorch.org/whl/cu128
 
24
  torchaudio==2.11.0+cu128
25
  mmcv-lite==2.2.0
 
21
  open_clip_torch
22
  git+https://github.com/descriptinc/audiotools
23
  --extra-index-url https://download.pytorch.org/whl/cu128
24
+ torch==2.11.0+cu128
25
  torchaudio==2.11.0+cu128
26
  mmcv-lite==2.2.0