CUDA IPC 业务
THCudaCheck FAIL file=torch\csrc\generic\StorageSharing.cpp line=252
error=63 : OS call failed or operation not supported on this OS
Windows 不支持此类业务。就像在 CUDA 张量上进行多进程处理一样无法成功,有两种方法可供选择:
- 不要使用多进程处理。将DataLoader 的num_worker设置为零。
- 改为 CPU 共享张量。确保您的自定义DataSet数据集返回 CPU 张量。