Developer
(파이썬에서) 특정 GPU만 사용하고 싶을 때
South Korea life
2021. 8. 1. 21:38
728x90
반응형
[ 제 네이버 블로그 글을 티스토리로 이전한 글입니다. 해당 글은 제 네이버 블로그에 2017.11.17에 작성되었었습니다 ]
컴퓨터에 여러 GPU를 사용할 경우, 그 중 특정 하나의 GPU만 사용하여 연산을 수행하고 싶을 때가 있다.
이 때, 유용하게 사용할 수 있는 command가 CUDA_VISIBLE_DEVICES이다. (시스템이 볼 수 있는 devices를 제한함)
사용 예) ubuntu$ CUDA_VISIBLE_DEVICES=x python ~~.py
728x90
반응형