运行pytorch程序时,因为中间有把数据转成cpu的操作,跑多卡时会把报错:RuntimeError: cuda runtime error (700) : an illegal memory access was encountered at /pytorch/aten/src/THC/THCCachingHostAllocator.cpp
直接改成跑单卡,或者把中间的cpu部分的运算全都改到gpu上。
【debug】RuntimeError: cuda runtime error (700) : an illegal memory access was encountered解决
最新推荐文章于 2024-03-02 15:36:43 发布