search

sp-uhh / storm

StoRM: A Diffusion-based Stochastic Regeneration Model for Speech Enhancement and Dereverberation
MIT License
164 stars 22 forks source link

Running on cpu only mode or with no CUDA gpu #10

Closed adeelabbas closed 1 year ago

adeelabbas commented 1 year ago

Hi When I run the code on my macbook pro (no cuda gpu), I get this:

Traceback (most recent call last):
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/enhancement.py", line 13, in <module>
    from sgmse.backbones.shared import BackboneRegistry
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/__init__.py", line 2, in <module>
    from .ncsnpp import AutoEncodeNCSNpp, NCSNpp, NCSNppLarge, NCSNpp12M, NCSNpp6M
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/ncsnpp.py", line 18, in <module>
    from .ncsnpp_utils import layers, layerspp, normalization
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/ncsnpp_utils/layerspp.py", line 20, in <module>
    from . import up_or_down_sampling
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/ncsnpp_utils/up_or_down_sampling.py", line 10, in <module>
    from .op import upfirdn2d
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/ncsnpp_utils/op/__init__.py", line 1, in <module>
    from .fused_act import FusedLeakyReLU, fused_leaky_relu
  File "/Users/adeelabbas/Downloads/audio-dereverb/storm/sgmse/backbones/ncsnpp_utils/op/fused_act.py", line 11, in <module>
    fused = load(
  File "/Users/adeelabbas/Library/Python/3.9/lib/python/site-packages/torch/utils/cpp_extension.py", line 1202, in load
    return _jit_compile(
  File "/Users/adeelabbas/Library/Python/3.9/lib/python/site-packages/torch/utils/cpp_extension.py", line 1425, in _jit_compile
    _write_ninja_file_and_build_library(
  File "/Users/adeelabbas/Library/Python/3.9/lib/python/site-packages/torch/utils/cpp_extension.py", line 1514, in _write_ninja_file_and_build_library
    extra_ldflags = _prepare_ldflags(
  File "/Users/adeelabbas/Library/Python/3.9/lib/python/site-packages/torch/utils/cpp_extension.py", line 1622, in _prepare_ldflags
    extra_ldflags.append(f'-L{_join_cuda_home("lib64")}')
  File "/Users/adeelabbas/Library/Python/3.9/lib/python/site-packages/torch/utils/cpp_extension.py", line 2125, in _join_cuda_home
    raise EnvironmentError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

Do you know what changes are needed to run it on non-gpu machines?