상세 컨텐츠

본문 제목

(파이썬에서) 특정 GPU만 사용하고 싶을 때

Developer

by 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
반응형

관련글 더보기

댓글 영역