As in the warning message, pytorch FSDP has known bug for linux kernel<5.5.0:
...
Detected kernel version 5.4.0, which is below the recommended minimum of 5.5.0; this can cause the process to hang. It is recommended to upgrade the kernel to the minimum version or higher.
...
cf)