gpu로 돌리기1 Yolov8 GPU로 학습시키기 개요 참고한 사이트: https://artiiicy.tistory.com/61 [Pytorch] 특정 GPU 사용하기 / 여러개의 multi GPU parallel 하게 사용하기 Neural network를 train을 하다보면 성능을 비교하기 위해 loss function, optimizer 등이 상이한 여러 가지 model을 돌려볼 일이 수도 없이 많다. 이 때, 한 장의 GPU만 있다면 어쩔 수 없지만 여러장의 GPU가 존 artiiicy.tistory.com 참고한 사이트2: https://docs.ultralytics.com/ko/modes/train/#apple-m1-and-m2-mps-training 코드 1: GPU로 돌리기 코드 2: 멀티 GPU 데이터 병렬 모드 코드 1 import o.. 2024. 3. 19. 이전 1 다음