Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

StyleGAN3 error before generating video using pre trained model. #624

Open
tashfeen51 opened this issue Nov 2, 2023 · 0 comments
Open

Comments

@tashfeen51
Copy link

(stylegan3) C:\stylegan3\stylegan3>python gen_images.py --outdir=out --trunc=1 --seeds=2 --network=https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl
Loading networks from "https://api.ngc.nvidia.com/v2/models/nvidia/research/stylegan3/versions/1/files/stylegan3-r-afhqv2-512x512.pkl"...
C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\scipy_init_.py:146: UserWarning: A NumPy version >=1.16.5 and <1.23.0 is required for this version of SciPy (detected version 1.24.3
warnings.warn(f"A NumPy version >={np_minversion} and <{np_maxversion}"
No CUDA runtime is found, using CUDA_HOME='C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8'
Traceback (most recent call last):
File "C:\stylegan3\stylegan3\gen_images.py", line 143, in
generate_images() # pylint: disable=no-value-for-parameter
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\click\core.py", line 1157, in call
return self.main(*args, **kwargs)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\click\core.py", line 1078, in main
rv = self.invoke(ctx)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\click\core.py", line 1434, in invoke
return ctx.invoke(self.callback, **ctx.params)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\click\core.py", line 783, in invoke
return __callback(*args, **kwargs)
File "C:\stylegan3\stylegan3\gen_images.py", line 108, in generate_images
G = legacy.load_network_pkl(f)['G_ema'].to(device) # type: ignore
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 899, in to
return self._apply(convert)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply
module._apply(fn)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply
module._apply(fn)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 570, in _apply
module._apply(fn)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 593, in apply
param_applied = fn(param)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\nn\modules\module.py", line 897, in convert
return t.to(device, dtype if t.is_floating_point() or t.is_complex() else None, non_blocking)
File "C:\Users\ECEN\anaconda3\envs\stylegan3\lib\site-packages\torch\cuda_init
.py", line 208, in _lazy_init
raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled

I am getting these errors when i try to use pretrained model, before generating a video.
I have also tried installing different CUDA drivers (11.6, 11.8) but getting the same error.
When i check the list "conda list" it doesn't shows me the Cuda integration with the torch.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant