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?
Hi When I run the code on my macbook pro (no cuda gpu), I get this:
Do you know what changes are needed to run it on non-gpu machines?