ScaledYOLOv4 icon indicating copy to clipboard operation
ScaledYOLOv4 copied to clipboard

Terminate called after throwing an instance of 'c10::Error'

Open seekFire opened this issue 2 years ago • 0 comments

是在Scanning images环节出现了这个错误的, 如图: cuda_error 我按照网上的指导把环境变量CUDA_LAUNCH_BLOCKING设为1,可是并没有用 我使用的GPU是三张NVIDIA GTX TITAN X,显存是12GB的 我要训练的模型是YOLOv4-P5,模型的参数量记得是70多M 请问这个错误是否是由于显存大小不够而导致的呢?

seekFire avatar Mar 05 '22 14:03 seekFire