Azure Zest

Linux에서 GPU 할당하기 리눅스에서 GPU를 정하는 것은 쉽다. 코드를 돌릴 때, 앞에 CUDA_VISIBLE_DEVICES 를 붙이기만 하면된다. 아래와 같이 숫자는 각각 0부터 하나씩 지정이 되어 있을테니, 숫자를 바꿔서 각각 GPU에서 다른 코드를 돌려볼 수 있다. $ CUDA_VISIBLE_DEICES=0 python main.py Windows에서 GPU 할당하기 기본적으로 윈도우에서는 리눅스 명령어가 먹히지 않으므로, 저 명령어가 먹히지 않는다. 따라서 실행시킬 코드 위에 아래 코드를 적어 놓는다. 당연히, 리눅스에서도 아래의 코드를 미리 적어놓으면 앞에 CUDA_VISIBLE_DEVICES를 붙이지 않고, 바로 GPU를 할당할 수 있다. 위와 마찬가지로 숫자를 다르게 써서, 선택할 수 ..

Windows에서 GPU를 사용하나 확인! Ubuntu에서 tensorflow를 사용할 때, gpu를 사용하는 지는 nvidia-smi를 이용해서 확인할 수 있었다. 하지만, Windows10에서는 저 명령어를 쓸 수 없기 때문에 다른 방법으로 확인해야 한다. 그 중 제일 간편한 방법에 대해 알아보자. 아래의 하단 바를 우클릭해서 작업 관리자를 들어간다. 작업관리자의 성능 탭을 눌러서 그래프로 gpu를 얼마나 사용하고 있는지 확인할 수 있다. 이 때, 성능 탭을 눌렀는데도 GPU가 아예 보이지가 않는다면, 드라이버가 제대로 설치가 안 되었거나 GPU를 사용을 하지 않는 상태일 수 있다. 내가 겪었던 오류에 대해 살펴보도록 하자. 그래픽 디바이스 드라이버 오류 코드 22 혹시, 그래픽 카드 드라이버를 알맞..