티스토리 뷰

반응형

 

gpustat -i 혹은 nvidia-smi 쳤을때는 정상적으로 잡히는데

 

CUDA_VISIBLE_DEVICES만 하면 안되는 경우가 생겼다면 꼭 이걸 먼저 체크해보자

 

CUDA_VISIBLE_DEVICES 뒤의 숫자를 0 , 1 , 2  3 등등 바꿔가면서, 뒤의 숫자랑 그래픽카드 번호랑 매핑이 맞춰져있는지 먼저 확인해보자...

 

위의 두 사진에서 보이는 그래픽 카드 순서가 CUDA_VISIBLE_DEVICES = "???"에 나오는 ???랑 숫자 순서가 다르다는 것... 

 

CUDA_VISIBLE_DEVICES = 1 로 하니깐 4번째 GPU가 돌아가는 현상을 볼수 있다 ㅋㅋㅋ

 

엔비디아!! ㅂㄷㅂㄷ

 

아니면 파이토치 문제인가..?

 

 

반응형
댓글
공지사항
최근에 올라온 글