runtime error

Exit code: 1. Reason: ��███████| 381M/381M [00:04<00:00, 88.9MB/s] Downloading... From (original): https://drive.google.com/uc?id=1N0MZSqPRJpLfP4mFQCS14ikrVSe8vQlL From (redirected): https://drive.google.com/uc?id=1N0MZSqPRJpLfP4mFQCS14ikrVSe8vQlL&confirm=t&uuid=53b033fe-df81-47c8-8d10-d4557b924d28 To: /home/user/app/stylegan2-pytorch/model_ir_se50.pth 0%| | 0.00/175M [00:00<?, ?B/s] 19%|█▉ | 33.6M/175M [00:01<00:04, 33.5MB/s] 43%|████▎ | 76.0M/175M [00:02<00:02, 37.2MB/s] 81%|████████ | 142M/175M [00:03<00:00, 50.1MB/s]  100%|██████████| 175M/175M [00:03<00:00, 50.3MB/s] Traceback (most recent call last): File "/home/user/app/app.py", line 47, in <module> from model import Generator File "/home/user/app/stylegan2-pytorch/model.py", line 11, in <module> from op import FusedLeakyReLU, fused_leaky_relu, upfirdn2d, conv2d_gradfix File "/home/user/app/stylegan2-pytorch/op/__init__.py", line 1, in <module> from .fused_act import FusedLeakyReLU, fused_leaky_relu File "/home/user/app/stylegan2-pytorch/op/fused_act.py", line 11, in <module> fused = load( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1314, in load return _jit_compile( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1721, in _jit_compile _write_ninja_file_and_build_library( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1810, in _write_ninja_file_and_build_library extra_ldflags = _prepare_ldflags( File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 1899, in _prepare_ldflags if (not os.path.exists(_join_cuda_home(extra_lib_dir)) and File "/usr/local/lib/python3.10/site-packages/torch/utils/cpp_extension.py", line 2416, in _join_cuda_home raise OSError('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: