728x90
반응형
print('Allocated:', round(torch.cuda.memory_allocated(0)/1024**3,1), 'GB', 'Cached:', round(torch.cuda.memory_cached(0)/1024**3,1), 'GB')
#or
MB = 1024.0 * 1024.0
memory=torch.cuda.max_memory_allocated() / MB
728x90
반응형
'사소한 Tip . 오류 해결법 > python' 카테고리의 다른 글
| [python]특정 경로 내 파일 존재 유무 확인 (0) | 2022.03.16 |
|---|---|
| [python] 인터넷 크롤링(지마켓 상품명, 가격) / Python Crawling product name, price from online shopping mall (0) | 2022.03.15 |
| [python, layout-parser] pdf에서 정보 뽑기/ Extracting information from PDF file (0) | 2022.01.11 |
| [python] File Copy, shutil filecopy (0) | 2022.01.06 |
| [python] pdb command / pdb 명령어 / python debugger 사용법 (0) | 2021.01.15 |
