Pytorch运行程序报错

RuntimeError: copy_if failed to synchronize: cudaErrorIllegalAddress

  • 报错信息
  • 报错原因
  • 修改

报错信息

最近在训练人脸识别的模型,在将单卡训练改为多卡训练时,程序报错:

RuntimeError: copy_if failed to synchronize: cudaErrorIllegalAddress: an illegal memory access was encountered

报错原因

人脸识别中使用了amsoftmax损失,改代码时,未能将该部分模型加入到cuda上

修改

将amsoftmax损失部分加入到cuda上,程序正常运行

你可能感兴趣的:(pytorch,深度学习)