()
| 515 | # Helper Methods for multiprocess attacks |
| 516 | # |
| 517 | def pytorch_multiprocessing_workaround(): |
| 518 | # This is a fix for a known bug |
| 519 | try: |
| 520 | torch.multiprocessing.set_start_method("spawn", force=True) |
| 521 | torch.multiprocessing.set_sharing_strategy("file_system") |
| 522 | except RuntimeError: |
| 523 | pass |
| 524 | |
| 525 | |
| 526 | def set_env_variables(gpu_id): |