nn.DataParallel gets stuck

How did you disable this feature?

EDIT: Answering my own question. I modified this in my BIOS. Will be different depending on your system/MoBo maybe. But in my case I have an ASRock TRX40 Creator, hit delete or F11 at startup and went to Advanced → AMD CBS → NBIO → there should be an Option for IOMMU, which I set to disabled.

1 Like

If using centos, check this AMD documentation

https://docs.amd.com/bundle/IOMMU-Advisory-for-Multi-GPU-Environments/page/IOMMU_Advisory_for_Multi-GPU_Environments.html