File "C:\ProgramData\Anaconda3\envs\yolov5lite\lib\site-packages\torch\nn\functional.py", line 2438, in batch_norm
return torch.batch_norm(
RuntimeError: CUDA out of memory. Tried to allocate 26.00 MiB (GPU 0; 8.00 GiB total capacity; 6.65 GiB already allocated; 0 bytes free; 6.69 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF
GPU是 nvidia 1080 显存 8G
batch-size 设置64 报GPU内存不足,调小就报下面的类型错误