CUDA error: CUBLAS_STATUS_INVALID_VALUE when calling

autonomousvision / stylegan-xl

[SIGGRAPH'22] StyleGAN-XL: Scaling StyleGAN to Large Diverse Datasets

MIT License

961 stars 113 forks source link

Closed aminullah6264 closed 1 year ago

aminullah6264 commented 1 year ago

I am trying to run gen_images.py with the following command

python gen_images.py --outdir=out --trunc=0.7 --seeds=10-15 --batch-sz 1 --class 4 \ --network=https://s3.eu-central-1.amazonaws.com/avg-projects/stylegan_xl/models/imagenet256.pkl

I am getting the following error, is this related to cuda version? I am using cuda11.4 with the packages installed with environment.yml:

aminullah6264 commented 1 year ago

Just updating to the latest PyTorch and cuda11.6 solved the issue.