NVlabs / stylegan3

Official PyTorch implementation of StyleGAN3
Other
6.28k stars 1.1k forks source link

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

Open tashfeen51 opened 8 months ago

tashfeen51 commented 8 months ago

(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.