Google Colab环境下neural_renderer_pytorch安装失败问题求助

报错基础信息 #

• 运行环境：Google Colab 配备 Tesla P100-PCIE-16GB GPU（UUID: GPU-415e05e7-6236-d11b-c02d-65f7fec9509f）的实例，预装 Python 3.6、PyTorch 1.7.0+cu101
• 执行的安装命令：!pip install neural_renderer_pytorch
• 报错表现：所有Python依赖校验完成后，在源码编译生成wheel包阶段直接失败，pip进程返回退出码1，无明确前置编译错误提示。

核心问题原因 #

• PyPI 托管的 neural_renderer_pytorch 1.1.3 版本存在原生编译兼容缺陷：编译逻辑未自动识别 Colab 内置的 CUDA 安装路径，同时未适配 Python 3.6 环境的编译语法，直接触发编译中断。
• 实例默认未预装编译流程必需的基础构建工具链，进一步加剧了兼容问题。

解决操作步骤 #

• 安装系统层编译依赖，执行命令：
  !apt-get update && apt-get install -y build-essential ninja-build
• 手动配置CUDA环境变量，确保编译流程可定位到本机CUDA资源：

  !export CUDA_HOME=/usr/local/cuda
  !export PATH=$PATH:/usr/local/cuda/bin
  !export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
  

• 清空pip缓存，开启详细日志模式重新执行安装：
  !pip install neural_renderer_pytorch --no-cache-dir -v
• 安装完成后执行导入校验，确认安装生效：
  import neural_renderer as nr

内容的提问来源于stack exchange，提问作者molo32_1