OSError: libcudart.so.10.2: cannot open shared object file: No such file or directory

🐛 Bug Report

I get this error with latest Demux version:

Successfully built demucs
Installing collected packages: demucs
Successfully installed demucs-3.0.1

and the Colab example notebook provided:

Going to separate the files:
/gdrive/MyDrive/demucs/82929740.mp3
With command:  python3 -m demucs.separate -o /gdrive/MyDrive/demucs_separated/ -n mdx_extra --mp3 --mp3-bitrate=320
Traceback (most recent call last):
  File "/usr/lib/python3.7/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.7/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/usr/local/lib/python3.7/dist-packages/demucs/separate.py", line 14, in <module>
    import torchaudio as ta
  File "/usr/local/lib/python3.7/dist-packages/torchaudio/__init__.py", line 1, in <module>
    from torchaudio import _extension  # noqa: F401
  File "/usr/local/lib/python3.7/dist-packages/torchaudio/_extension.py", line 27, in <module>
    _init_extension()
  File "/usr/local/lib/python3.7/dist-packages/torchaudio/_extension.py", line 21, in _init_extension
    torch.ops.load_library(path)
  File "/usr/local/lib/python3.7/dist-packages/torch/_ops.py", line 110, in load_library
    ctypes.CDLL(path)
  File "/usr/lib/python3.7/ctypes/__init__.py", line 364, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libcudart.so.10.2: cannot open shared object file: No such file or directory
Command failed, something went wrong.

[adefossez]

that’s amazing! good thing they fixed it upstream.

[adefossez]

This is super weird. On my colab cuda 111 is definitely used. So maybe I need a script to detect the cuda version and install the right pytorch.