Open honghaier250 opened 4 months ago
Traceback (most recent call last): File "/kaggle/working/HairCLIPv2/./test.py", line 17, in from utils.model_utils import load_base_models File "/kaggle/working/HairCLIPv2/utils/model_utils.py", line 2, in from models.stylegan2.model import Generator File "/kaggle/working/HairCLIPv2/models/stylegan2/model.py", line 11, in from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "/kaggle/working/HairCLIPv2/models/stylegan2/op/init.py", line 1, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/kaggle/working/HairCLIPv2/models/stylegan2/op/fused_act.py", line 10, in fused = load( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1284, in load return _jit_compile( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1508, in _jit_compile _write_ninja_file_and_build_library( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1600, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1708, in _prepare_ldflags extra_ldflags.append(f'-L{_join_cuda_home("lib64")}') File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2230, 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.
I had the same problem, have you solved it yet?
Traceback (most recent call last): File "/kaggle/working/HairCLIPv2/./test.py", line 17, in from utils.model_utils import load_base_models File "/kaggle/working/HairCLIPv2/utils/model_utils.py", line 2, in from models.stylegan2.model import Generator File "/kaggle/working/HairCLIPv2/models/stylegan2/model.py", line 11, in from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "/kaggle/working/HairCLIPv2/models/stylegan2/op/init.py", line 1, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/kaggle/working/HairCLIPv2/models/stylegan2/op/fused_act.py", line 10, in fused = load( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1284, in load return _jit_compile( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1508, in _jit_compile _write_ninja_file_and_build_library( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1600, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1708, in _prepare_ldflags extra_ldflags.append(f'-L{_join_cuda_home("lib64")}') File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2230, 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.
I had the same problem, have you solved it yet?
No
Traceback (most recent call last): File "/kaggle/working/HairCLIPv2/./test.py", line 17, in from utils.model_utils import load_base_models File "/kaggle/working/HairCLIPv2/utils/model_utils.py", line 2, in from models.stylegan2.model import Generator File "/kaggle/working/HairCLIPv2/models/stylegan2/model.py", line 11, in from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d File "/kaggle/working/HairCLIPv2/models/stylegan2/op/init.py", line 1, in from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/kaggle/working/HairCLIPv2/models/stylegan2/op/fused_act.py", line 10, in fused = load( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1284, in load return _jit_compile( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1508, in _jit_compile _write_ninja_file_and_build_library( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1600, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1708, in _prepare_ldflags extra_ldflags.append(f'-L{_join_cuda_home("lib64")}') File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2230, 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.
I had the same problem, have you solved it yet?
Do you have any idea ?
@wty-ustc
Traceback (most recent call last): File "/kaggle/working/HairCLIPv2/./test.py", line 17, in
from utils.model_utils import load_base_models
File "/kaggle/working/HairCLIPv2/utils/model_utils.py", line 2, in
from models.stylegan2.model import Generator
File "/kaggle/working/HairCLIPv2/models/stylegan2/model.py", line 11, in
from models.stylegan2.op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d
File "/kaggle/working/HairCLIPv2/models/stylegan2/op/init.py", line 1, in
from .fused_act import FusedLeakyReLU, fused_leaky_relu
File "/kaggle/working/HairCLIPv2/models/stylegan2/op/fused_act.py", line 10, in
fused = load(
File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1284, in load
return _jit_compile(
File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1508, in _jit_compile
_write_ninja_file_and_build_library(
File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1600, in _write_ninja_file_and_build_library
extra_ldflags = _prepare_ldflags(
File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1708, in _prepare_ldflags
extra_ldflags.append(f'-L{_join_cuda_home("lib64")}')
File "/opt/conda/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2230, 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.