CUDA error on WSL2 using pytorch with multiprocessing – Ubuntu
I have a Python script as shown below: import torch from torch.multiprocessing import set_start_method, Pipe, Process def func(conn): data = conn.recv() print(data) if __name__ == "__main__": set_start_method('spawn') a, b = Pipe() data = torch.tensor([1, 2, 3], device='cuda') proc = Process(target=func,…