pytorch2 파이토치 GPU 사용하는 방법 파이토치 환경에서 학습을 하다보면 학습 속도가 매우 느릴 때가 있다. 그럴 때 작업 관리자를 켜고 보면 GPU를 사용하지 않고 CPU 사용할 때가 매우 많다. 그래서 이번엔 파이토치를 사용하여 학습을 진행할 때 GPU를 제대로 인식하고 사용하는지에 대한 여부를 확인하는 방법에 대해 기술하려고 한다. (파이토치가 설치되었다는 가정 하에 글을 진행한다.) 당연히 토치는 임포트 돼있어야한다. GPU 사용 가능 여부 print(torch.cuda.is_available()) GPU가 제대로 인식되어 있고 토치가 설치되어 있고 gpu를 사용할 수 있다면 True, 그렇지 않으면 False가 출력된다. 사용 가능한 GPU 개수 print(torch.cuda.device_count()) 위 명령어를 입력하면 사용 가.. 2024. 1. 16. AttributeError: 'tuple' object has no attribute 'size' 파이토치를 이용해 학습을 하려는 도중에 아래와 같은 오류가 발생했다. 오류메세지: AttributeError: 'tuple' object has no attribute 'size' 나는 현재 BCEWithLogitsLoss 손실함수를 사용하며 이진분류를 진행하고 있다. 이 에러는 input과 target의 크기가 다른 것으로, 둘 중 하나가 튜플(tuple) 형태로 되어있어 .size() 메소드를 호출할 수 없다는 것을 의미한다. for epoch in range(1, args['epoch'] + 1): model.train() running_corrects = 0 for videos, labels in tqdm(iter(train_loader), leave=False, desc='Train', asci.. 2024. 1. 11. 이전 1 다음
