Pytorch Windows不支持CUDA IPC操作?有解决办法吗?

Pytorch Windows不支持CUDA IPC操作?有解决办法吗?,pytorch,Pytorch,尝试使用自定义数据加载器训练神经网络时: train_dataset = datasets.VectorizedDataset(x_train,y_train) train_loader = datasets.DataLoader(shuffle=True,num_workers=4) 这会在加载数据时立即出错: RuntimeError: cuda runtime error (71) : operation not supported at c:\a\w\1\s\tmp_conda_3.6

尝试使用自定义数据加载器训练神经网络时:

train_dataset = datasets.VectorizedDataset(x_train,y_train)
train_loader = datasets.DataLoader(shuffle=True,num_workers=4)
这会在加载数据时立即出错:

RuntimeError: cuda runtime error (71) : operation not supported at c:\a\w\1\s\tmp_conda_3.6_090826\conda\conda-bld\pytorch_1550394668685\work\torch\csrc\generic\StorageSharing.cpp:232
本文件(CUDA IPC ops)记录如下:

Windows不支持它们。像做什么 CUDA张量上的多处理无法成功,有两种方法 这方面的替代方案

  • 不要使用多重处理。将DataLoader的num_worker设置为零

  • 共享CPU张量。确保自定义数据集返回CPU张量

  • 我通过PyTorch 1.0.1和CUDA 10.0在Windows 10上使用GTX 1070 GPU

    在Pytorch中,确实不可能将CUDA tensor数据加载卸载到Windows机器上的多个CPU上吗?有解决办法吗?我们是否必须做一些更复杂的事情,比如编写自己的多线程?还是我们被迫将所有数据加载到主进程上

    谢谢你的帮助