我能够在具有CUDA 9.2,Quadro M5000(5)和python 3.7.0(anaconda)的系统上运行我的程序,而同一程序使我在具有CUDA 10.0,GeForce GTX 1080 Ti的另一个系统上出现错误(6)和python 3.7.1
这是错误
ForkingPickler(file, protocol).dump(obj)
BrokenPipeError: [Errno 32] Broken pipe
整个代码都包装在一个名为train_valid_model()的函数中,只需在脚本中添加主要的保护措施即可:
def train_valid_model():
# complete code here
if __name__=='__main__':
train_valid_model()
它仅在num_workers = 0时有效,如果将其设置为1或2,则会出现此错误。知道为什么在程序相同的情况下我会收到此错误,只是CUDA版本和python版本不同吗?