StyleGAN3卡在Setting up PyTorch plugin 'filtered_lrelu_plugin'求助

1. 清理PyTorch插件缓存 #

Windows下PyTorch的插件缓存易因强制中断损坏，直接删除以下路径的缓存文件夹：

• C:\Users\<你的用户名>\AppData\Local\torch_extensions\torch_extensions\Cache
• StyleGAN3项目目录下的torch_extensions文件夹也一并删除

2. 强制重新编译插件 #

运行生成图片命令时添加强制重载参数，跳过损坏的缓存：

python gen_images.py --force_reload --outdir=out --trunc=1 --seeds=0-3 --network=network-snapshot.pkl

--force_reload会强制PyTorch重新编译filtered_lrelu_plugin，避免复用错误缓存。

3. 终止残留进程 #

强制中断后可能有未关闭的进程占用资源：

• 打开任务管理器，结束所有python.exe、pythonw.exe进程
• 用taskkill /F /PID <进程ID>命令终止nvidia-smi中显示的残留CUDA进程

4. 验证CUDA与Torch版本兼容性 #

版本不匹配可能导致插件编译失败：

• 运行python -c "import torch; print(torch.version.cuda)"查看Torch内置CUDA版本
• 确保系统安装的CUDA Toolkit版本与上述版本一致，或降级到兼容版本

5. 临时禁用插件加速（应急方案） #

若以上方法无效，可修改代码禁用自定义插件：

• 打开stylegan3/torch_utils/ops/filtered_lrelu.py，将use_plugin = True改为use_plugin = False
• 此方法会用纯PyTorch实现替代CUDA插件，速度会变慢，但能验证是否为插件问题

内容的提问来源于stack exchange，提问作者Игорь Платонов