使用音频算子报错：ImportError: cannot import name 'AudioDecoder' from 'torchcodec.decoders' (/usr/local/lib/python3.10/dist-packages/torchcodec/decoders/__init__.py)

[Young-zj]

Before Asking 在提问之前

• I have read the README carefully. 我已经仔细阅读了 README 上的操作指引。

• I have pulled the latest code of main branch to run again and the problem still existed. 我已经拉取了主分支上最新的代码，重新运行之后，问题仍不能解决。

Search before asking 先搜索，再提问

• I have searched the Data-Juicer issues and found no similar questions. 我已经在 issue列表 中搜索但是没有发现类似的问题。

Question

我在使用 audio 系列算子（audio_duration_filter, audio_add_gaussian_noise_mapper, audio_nmf_snr_filter 等）时，遇到持续可复现的导入错误：

ImportError: cannot import name 'AudioDecoder' from 'torchcodec.decoders'

报错堆栈来自内部的：

datasets.features.audio.Audio.encode_example
→ from torchcodec.decoders import AudioDecoder

但是我已经下载了正确版本的 torchcodec ，还是报这个错误，这些audio有什么其他运行条件吗？

Additional 额外信息

No response

[HYLcool]

嗨，请问你安装的是哪一个版本的torch和torchcodec？目前最新版本data-juicer中需要torch==2.8.0，torchcodec==0.7，其中0.7版本中的torchcodec.decoders中是有AudioDecoder的
https://github.com/meta-pytorch/torchcodec/blob/7dd6092b40a76a262b633c591d8edb6ce0a86c11/src/torchcodec/decoders/__init__.py#L8

[Young-zj]

嗨，请问你安装的是哪一个版本的torch和torchcodec？目前最新版本data-juicer中需要torch==2.8.0，torchcodec==0.7，其中0.7版本中的torchcodec.decoders中是有AudioDecoder的 https://github.com/meta-pytorch/torchcodec/blob/7dd6092b40a76a262b633c591d8edb6ce0a86c11/src/torchcodec/decoders/__init__.py#L8

这个是我的dj版本和torch版本
Data-Juicer version: 1.4.3
Torch version: 2.6.0+cu124
TorchCodec version: 0.2.1+cu124

[HYLcool]

嗨，请问你安装的是哪一个版本的torch和torchcodec？目前最新版本data-juicer中需要torch==2.8.0，torchcodec==0.7，其中0.7版本中的torchcodec.decoders中是有AudioDecoder的 https://github.com/meta-pytorch/torchcodec/blob/7dd6092b40a76a262b633c591d8edb6ce0a86c11/src/torchcodec/decoders/__init__.py#L8

这个是我的dj版本和torch版本 Data-Juicer version: 1.4.3 Torch version: 2.6.0+cu124 TorchCodec version: 0.2.1+cu124

可以尝试手动将torchcodec升级到0.3.0及以上版本，0.2.1版本中torchcodec.decoders是没有AudioDecoder的

[Young-zj]

嗨，请问你安装的是哪一个版本的torch和torchcodec？目前最新版本data-juicer中需要torch==2.8.0，torchcodec==0.7，其中0.7版本中的torchcodec.decoders中是有AudioDecoder的 https://github.com/meta-pytorch/torchcodec/blob/7dd6092b40a76a262b633c591d8edb6ce0a86c11/src/torchcodec/decoders/__init__.py#L8

这个是我的dj版本和torch版本 Data-Juicer version: 1.4.3 Torch version: 2.6.0+cu124 TorchCodec version: 0.2.1+cu124

可以尝试手动将torchcodec升级到0.3.0及以上版本，0.2.1版本中torchcodec.decoders是没有AudioDecoder的

感谢提示。这个音频算子我是运行在官方镜像里面的，最新dj官方镜像(基于 CUDA 12.4.1)，而它只能支持到toch 2.6 而torch 2.6 对应的 torchcodec 版本是 0.2，如果强行在官方镜像安装 torch 2.8 和 torchcode 0.7 就会因为 cuda 版本不匹配，官方镜像其实可以用cuda126为基础层，这样就可以避免cuda，torch，torchcode三角依赖冲突。

[HYLcool]

感谢你的建议👍🏻目前torch和torchcodec的版本要求还在最新的主分支中，将更新于下个小版本里，到时候会对pypi包和docker镜像都进行同步更新，我们会参照你的建议对基础镜像也进行更新~