Make torchaudio loading optional when compiling for AMD

Summary: TSIA

Differential Revision: D57803200

fbshipit-source-id: e45fef35243258cccd41b8bdca6b0893fab94a3d

mmf/datasets/processors/image_processors.py

@@ -41,12 +41,13 @@ def __init__(self, config, *args, **kwargs):
 
             transform = getattr(transforms, transform_type, None)
             if transform is None:
-                from mmf.utils.env import setup_torchaudio
+                if not (torch.cuda.is_available() and torch.version.hip):
+                    from mmf.utils.env import setup_torchaudio
 
-                setup_torchaudio()
-                from torchaudio import transforms as torchaudio_transforms
+                    setup_torchaudio()
+                    from torchaudio import transforms as torchaudio_transforms
 
-                transform = getattr(torchaudio_transforms, transform_type, None)
+                    transform = getattr(torchaudio_transforms, transform_type, None)
             # If torchvision or torchaudiodoesn't contain this, check our registry
             # if we implemented a custom transform as processor
             if transform is None: