在导入torchaudio的时候报错

import torchaudio

报错如下:

  File "a.py", line 1, in <module>
    import torchaudio
  File "/root/miniconda3/envs/wetts/lib/python3.8/site-packages/torchaudio/__init__.py", line 1, in <module>
    from torchaudio import _extension  # noqa: F401
  File "/root/miniconda3/envs/wetts/lib/python3.8/site-packages/torchaudio/_extension.py", line 67, in <module>
    _init_extension()
  File "/root/miniconda3/envs/wetts/lib/python3.8/site-packages/torchaudio/_extension.py", line 61, in _init_extension
    _load_lib("libtorchaudio")
  File "/root/miniconda3/envs/wetts/lib/python3.8/site-packages/torchaudio/_extension.py", line 51, in _load_lib
    torch.ops.load_library(path)
  File "/root/miniconda3/envs/wetts/lib/python3.8/site-packages/torch/_ops.py", line 643, in load_library
    ctypes.CDLL(path)
  File "/root/miniconda3/envs/wetts/lib/python3.8/ctypes/__init__.py", line 373, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: libtorch_cuda_cpp.so: cannot open shared object file: No such file or directory

原因:torch的版本和torchaudio的版本对应不上

解决方法:

卸载torchaudio,重新安装最新版本

pip uninstall torchaudio
pip install torchaudio

或者:卸载torch,重新安装最新版本

pip uninstall torch
pip install torch

或者:如果对torch有指定版本的要求,则在官网找到对应版本的torch进行重新按照,官网:https://pytorch.org/get-started/previous-versions/
例如,要安装torch==2.0.0,cuda版本为11.7,则:

pip install torch==2.0.0+cu117 torchvision==0.15.1+cu117 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu117

相应的安装命令直接在上述网址能直接复制

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐