※ 본 글은 필자의 경험에 근거하여 작성되었으며, 잘못된 정보가 포함될 수 있습니다. ※
딥러닝을 하다 보면 Jupyter Notebook이나 Jupyter-lab을 사용하여 코드를 작성하는 경우가 많다.
다만, 원하는 GPU만 선택하여 사용하고 싶은 경우가 있는데, 아래의 코드를 가장 위쪽 부분에 넣어주면 된다. 혹시나 적용되지 않는 경우에는, 커널 재시작(kernel restart)을 진행한 뒤에 다시 주피터 노트북 코드를 실행해 보자.
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"