여러 개의 GPU중 일부만 선택하여 사용하기
2017. 8. 1. 14:17
서버에 여러 개의 GPU가 장착되어 있고, 여러 사람이 접속하여 자신의 일을 해야하는 경우 모든 GPU가아니라 일부를 사용해야하는 경우가 있습니다.
아래의 코드는 GPU 번호 3,4번을 사용하는 코드입니다. nvidia-smi를 통해 몇 번 GPU가 사용중인지를 체크하고 비어있는 GPU를 사용할 수 있습니다.
import os
os.environ['CUDA_VISIBLE_DEVICES'] = '3, 4'