lucidrains / audiolm-pytorch

Implementation of AudioLM, a SOTA Language Modeling Approach to Audio Generation out of Google Research, in Pytorch
MIT License
2.32k stars 249 forks source link

Error when running 3rd cell in demo ipynb #229

Closed uday18git closed 10 months ago

uday18git commented 10 months ago

FileNotFoundError Traceback (most recent call last) Input In [4], in <cell line: 8>() 6 import urllib.request 7 import tarfile ----> 8 from audiolm_pytorch import SoundStream, SoundStreamTrainer, HubertWithKmeans, SemanticTransformer, SemanticTransformerTrainer, HubertWithKmeans, CoarseTransformer, CoarseTransformerWrapper, CoarseTransformerTrainer, FineTransformer, FineTransformerWrapper, FineTransformerTrainer, AudioLM 9 from torch import nn 10 import torch

File D:\audiolm-pytorch\audiolm_pytorch__init__.py:8, in 5 from einops._torch_specific import allow_ops_in_compiled_graph 6 allow_ops_in_compiled_graph() ----> 8 from audiolm_pytorch.audiolm_pytorch import AudioLM 9 from audiolm_pytorch.soundstream import SoundStream, AudioLMSoundStream, MusicLMSoundStream 10 from audiolm_pytorch.encodec import EncodecWrapper

File D:\audiolm-pytorch\audiolm_pytorch\audiolm_pytorch.py:13, in 10 import torch.nn.functional as F 11 from torch.nn.utils.rnn import pad_sequence ---> 13 import torchaudio 15 from einops import rearrange, repeat, reduce 16 from einops.layers.torch import Rearrange

File D:\Anaconda\lib\site-packages\torchaudio__init__.py:1, in ----> 1 from torchaudio import ( # noqa: F401 2 _extension, 3 compliance, 4 datasets, 5 functional, 6 io, 7 kaldi_io, 8 models, 9 pipelines, 10 sox_effects, 11 transforms, 12 utils, 13 ) 14 from torchaudio.backend import get_audio_backend, list_audio_backends, set_audio_backend 16 try:

File D:\Anaconda\lib\site-packages\torchaudio_extension.py:135, in 127 if ta_version != t_version: 128 raise RuntimeError( 129 "Detected that PyTorch and TorchAudio were compiled with different CUDA versions. " 130 f"PyTorch has CUDA version {t_version} whereas TorchAudio has CUDA version {ta_version}. " 131 "Please install the TorchAudio version that matches your PyTorch version." 132 ) --> 135 _init_extension() 136 _check_cuda_version()

File D:\Anaconda\lib\site-packages\torchaudio_extension.py:105, in _init_extension() 102 except Exception: 103 pass --> 105 _load_lib("libtorchaudio") 106 # This import is for initializing the methods registered via PyBind11 107 # This has to happen after the base library is loaded 108 from torchaudio import _torchaudio # noqa

File D:\Anaconda\lib\site-packages\torchaudio_extension.py:52, in _load_lib(lib) 50 if not path.exists(): 51 return False ---> 52 torch.ops.load_library(path) 53 torch.classes.load_library(path) 54 return True

File D:\Anaconda\lib\site-packages\torch_ops.py:255, in _Ops.load_library(self, path) 250 path = torch._utils_internal.resolve_library_path(path) 251 with dl_open_guard(): 252 # Import the shared library into the process, thus running its 253 # static (global) initialization code in order to register custom 254 # operators with the JIT. --> 255 ctypes.CDLL(path) 256 self.loaded_libraries.add(path)

File D:\Anaconda\lib\ctypes__init.py:382, in CDLL.init__(self, name, mode, handle, use_errno, use_last_error, winmode) 379 self._FuncPtr = _FuncPtr 381 if handle is None: --> 382 self._handle = _dlopen(self._name, mode) 383 else: 384 self._handle = handle

FileNotFoundError: Could not find module 'D:\Anaconda\Lib\site-packages\torchaudio\lib\libtorchaudio.pyd' (or one of its dependencies). Try using the full path with constructor syntax.