StyleGAN-xl Setting up PyTorch plugin “bias_act_plugin“... Failed或“upfirdn2d_plugin“... Failed

问题

  File "<frozen importlib._bootstrap>", line 565, in module_from_spec
  File "<frozen importlib._bootstrap_external>", line 1173, in create_module
  File "<frozen importlib._bootstrap>", line 228, in _call_with_frames_removed
ImportError: DLL load failed while importing bias_act_plugin: 找不到指定的模块。

查看环境cuda版本和系统cuda版本

# 检查PyTorch CUDA版本
python -c "import torch; print('PyTorch CUDA version:', torch.version.cuda)"
# 检查CUDA版本
nvcc --version

发现系统cuda=12.2，StyleGAN推进pytorch1.9.1（cuda=11.1）

删除build目录（若存在），并重新运行setup.py。使用以下命令强制重新编译：

# 系统
nvcc: NVIDIA (R) Cuda compiler driver
Cuda compilation tools, release 12.2, V12.2.140
Build cuda_12.2.r12.2/compiler.33191640_0
# 环境
PyTorch CUDA version: 11.1

卸载Cuda，删除torch_extensions缓存，重新安装cuda11.1

https://developer.nvidia.com/cuda-11.1.1-download-archive

正常加载plugin，开始训练