전체 글 썸네일형 리스트형 2025년 2월 6일 토스 1등찍기 테팔 후라이팬 더보기 2025년 2월 5일 토스 1등찍기 오렌지 더보기 2025년 2월 4일 토스 1등찍기 마미앙 쇼파 더보기 2025년 2월 3일 토스 1등찍기 1등은 롱패딩 더보기 2025년 2월 2일 토스 1등찍기 1등 마이밀 뉴프로틴 더보기 2025년 2월 1일 토스 1등찍기 1등은 세노키즈 비타민 더보기 2025년 1월 28일 토스 1등찍기 토스 1등 투표는 현재 식판과 행주가 앞서갑니다. 더보기 pytorch에서 gpu 슬롯 별 메모리 여유량 알아보는 기능 1. 개요gpu사용량을 파이썬 내에서 실시간으로 알아내면 모델 학습시에 편리하기 때문에 아래의 기능을 만듬.먼저 print기능을 통해 gpu 슬롯 별 여유량을 출력하게 만듬.그 후에 파이썬의 리스트에 각 슬롯번호 순서대로 메모리 사용량을 저장하게 만듬.import torchdef get_gpu_memory_usage(): memories=[] for i in range(torch.cuda.device_count()): free, total = torch.cuda.mem_get_info(f'cuda:{i}') mem_used_mb = (total - free) / 1024 ** 2 print(f'cuda:{i} free :',free/1024**3,'Gb').. 더보기 이전 1 2 3 4 5 ··· 109 다음
