강제로 GPU 를 미사용하고 싶을 때

 

가끔은 GPU 사용을 안하고 CPU 모드로만 동작해서 테스트할 경우가 있습니다.

이때 소스코드 수정 없이 쉽게 GPU 를 미사용하게 만드는 방법이 있습니다.

 

가장 쉬운 방법으로 환경변수 CUDA_VISIBLE_DEVICES 의 값을 "" 처럼 빈값으로 처리하면 됩니다.

 

만약, 리눅스의 경우 다음과 같이 입력하면 됩니다.

export CUDA_VISIBLE_DEVICES=""

 

 

CUDA_VISIBLE_DEVICES 라는 환경변수는 GPU 를 지정해서 사용할 수 있게 도와주는 환경변수입니다.

여기에 "" 처럼 아무값도 넣지 않는다면 GPU를 사용할 수 없는 것으로 인식되게 됩니다.

따라서 이렇게 한줄만 입력하면, GPU 없는 상태로 강제로 테스트가 가능해집니다.

 

만약 다시 GPU 를 사용하고 싶다고 하면 아래와 같은 방법으로 GPU 번호를 직접 지정해서 사용할 수 있습니다.

export CUDA_VISIBLE_DEVICES="0"  #0번만 사용할 경우 
export CUDA_VISIBLE_DEVICES="1,2" #1번, 2번 사용할 경우

 

그럼 즐거운 개발 되세요.