제가 yolov5를 모델을 통해 spalling 자국을 알아내는 모형을 할려고 합니다. 제가 사용한 이미지의 크기가 대개 3587*10000픽셀로 크기가 되게 큰 편입니다. 그래서 코드를
!python train.py --img 3616 --batch 1 --workers 2 --epoch 100 --data C:/Users/User1015/content/yolov5/data.yaml --cfg C:/Users/User1015/content/yolov5/models/yolov5l.yaml --weights yolov5l.pt --name road_spalling
넣고 돌리는데 torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 50.00 MiB (GPU 0; 8.00 GiB total capacity; 7.16 GiB already allocated; 0 bytes free; 7.27 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 이 에러가 뜨네요.
그래서 방법을 찾아서 넣으면
OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.이게 나와서 계속 둘이 번갈아가면서 에러코드가 뜹니다! 이걸 어떻게 해결해야 되나요?
저는 참고로
tensorflow-gpu==2.10.0
cuda == 11.2
cuDNN == 8.1
NVIDIA GeForce RTX 2070 SUPER 를 사용합니다.