using python3 and adding this to the absolute top of your main script will help fix the forking issues:
import torch
if __name__ == "__main__":
    import torch.multiprocessing.set_start_method("spawn")