Pytorch 에서 gpu 사용률 늘리기

반응형

4080이라는 좋은 그래픽 카드로 바꾸고 데스크탑도 새로 맞췄는데 

그렇게만 하면 되는 줄 알았다. 

 

윈도우에서 우분투 환경을 세팅하는데 4일이 넘게 걸렸다...

결국 wsl2 에서 우분투 환경을 만들어주고 프로젝트 과제를 하는데 gpu를 전혀 사용하지 못하고 있다.

 

위의 사진을 보면 Volatile이라고 적혀있는 가장 오른쪽 부분의 % 가 gpu를 얼마나 사용하는지에 대한 내용이라고 하는데 

보다시피 2%에 머물러있다. 

 

1. 배치 사이즈 문제 
2. Dataloader를 설정할 때 num_worker를 늘려주기. 보통 gpu 개수 * 4로 한다고 한다.

위의 방법은 학습을 돌릴 때 신경을 써야하는 부분.

 

앞으로 실시간 gpu 사용량을 보며 학습을 돌려야겠다...

반응형