【debug】RuntimeError: cuda runtime error (700) : an illegal memory access was encountered解决

运行pytorch程序时,因为中间有把数据转成cpu的操作,跑多卡时会把报错:RuntimeError: cuda runtime error (700) : an illegal memory access was encountered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp
直接改成跑单卡,或者把中间的cpu部分的运算全都改到gpu上。

你可能感兴趣的:(pytorch)