728x90
resnet152 test
resnet152를 사용하여 환경별 GPU 사용량 테스트
NVIDIA Environment
- nvidia gpu driver version : 470.103.01
- cuda version : 11.4
- gpu device : NVIDIA GeForce RTX 3090
- NVIDIA GeForce RTX 3090
- memory size : 24GB
- cuda : 8.5
- cuda core : 104996
- rt core : 82
- tensor core : 328
etc
- torch version : 2.0.0
- torch cuda version : 11.7
- resnet152 model
- file size : 230.4 MB
- num_params : 60,192,808
- test image : dog.jpg
테스트 케이스 4가지
- normal ( eval x & no_grad x ) : 2254MiB
- model.eval() : 2228MiB
- model.no_grad() : 1896MiB
- model.eval() & model.no_grad() : 1870MiB
test case image
- normal ( eval x & no_grad x )
- model.eval()
- model.no_grad()
- model.eval() & model.no_grad()
728x90
'Machine Learning > A·I' 카테고리의 다른 글
선형회귀모델 (0) | 2022.08.15 |
---|---|
머신러닝 모델 프로세스 (0) | 2022.08.14 |
수치 예측, 범주 예측 (분류) (0) | 2022.08.13 |
머신러닝, 인공지능이란? (0) | 2022.08.12 |
에러, fastai book import error (2) | 2022.03.31 |