반응형
반응형
pin_memory란?
pin_memory는 PyTorch에서 제공하는 옵션으로써, GPU에서 데이터 로드할 때 빠르게 옮기기 위해서 사용하는 것을 말합니다.
pin_memory 사용하는 이유는?
pin_memor를 사용하는 이유는 GPU 같은 경우 DRAM에 할당된 데이터(변수)를 사용하지 못하기 때문에 GPU에서 사용하기 위해 pinned memory로 데이터롤 옮기고 다시 VRAM(GPU 메모리)로 옮기는 과정에서 속도가 많이 느려지는 것을 막기 위해 DRAM을 통하지 않고 바로 VRAM으로 데이터 할당해줘 데이터 로드 속도 개선하는 역활을 하기 때문입니다.
반응형
'머신러닝 & 딥러닝' 카테고리의 다른 글
| ModuleNotFoundError: No module named "" (0) | 2023.04.25 |
|---|---|
| 'NoneType' object has no attribute 'shape' (0) | 2023.04.13 |
| torchvision.transforms.ToTensor() (0) | 2023.04.10 |
| Data annotation이란? (0) | 2023.04.10 |
| end-to-end란? (0) | 2023.04.09 |