jetson-containers icon indicating copy to clipboard operation
jetson-containers copied to clipboard

NeMo doesn't support Jetson.

Open UserName-wang opened this issue 2 years ago • 2 comments
trafficstars

just let you know that NeMo doesn't support Jetson. https://github.com/NVIDIA/NeMo/issues/7331

but what's the docker image used for? dustynv/nemo:r35.3.1

UserName-wang avatar Aug 28 '23 09:08 UserName-wang

@UserName-wang you are correct that Nemo doesn't officially support Jetson - however I was able to get it building with some tweaks and am able to use it for some things. The stuff in Nemo that requires torch.distributed doesn't work (like megatron)

dusty-nv avatar Aug 28 '23 13:08 dusty-nv

@UserName-wang you are correct that Nemo doesn't officially support Jetson - however I was able to get it building with some tweaks and am able to use it for some things. The stuff in Nemo that requires torch.distributed doesn't work (like megatron)

but I even cannot complete: import nemo.collections.asr as nemo_asr here is the error message: https://github.com/NVIDIA/NeMo/files/12450309/error_message.txt I just need TTS, ASR.

UserName-wang avatar Aug 28 '23 14:08 UserName-wang