在Google Colab微调NeMo Parakeet遇CUDA PTX版本不兼容问题求助

要让你找到的修复方案在整个NeMo库中全局生效，只需在Notebook的最开头（所有导入语句和NeMo初始化操作之前）执行以下步骤，无需逐个模块单独设置：

• 先安装指定版本的依赖包，避免版本冲突：

!pip install numba-cuda==0.4.0 numpy==1.26.4

• 设置全局环境变量，确保numba加载时读取正确的PTX架构版本：

import os
os.environ["NUMBA_CUDA_DEFAULT_PTX_CC"] = "7.5"

• 全局开启numba的pynvjitlink支持：

from numba import config
config.CUDA_ENABLE_PYNVJITLINK = 1

完成以上操作后，再导入NeMo相关模块并初始化模型。这些配置会全局作用于所有依赖numba的NeMo组件——环境变量和numba的config设置是进程级别的，NeMo内部调用numba时会自动继承这些参数。

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