CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity

pytorch的yolov4

  • CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity
    • 解决办法

CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 6.00 GiB total capacity

pytorch的yolov4,训练过程出现cuda out of memory

解决办法

windows下将num_workers修改为1,就可以了

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