runtime error

Traceback (most recent call last): File "/home/user/app/app.py", line 11, in <module> from models.stylegene.api import synthesize_descendant File "/home/user/app/models/stylegene/api.py", line 4, in <module> from models.stylegan2.model import Generator File "/home/user/app/models/stylegan2/model.py", line 7, in <module> from .op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "/home/user/app/models/stylegan2/op/__init__.py", line 1, in <module> from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/home/user/app/models/stylegan2/op/fused_act.py", line 10, in <module> fused = load( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1202, in load return _jit_compile( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1425, in _jit_compile _write_ninja_file_and_build_library( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1514, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/home/user/.local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1622, in _prepare_ldflags extra_ldflags.append(f'-L{_join_cuda_home("lib64")}') File "/home/user/.local/lib/python3.10/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.

Container logs: