728x90
반응형
가끔 GPU를 나눠 할당하여 코드를 실행하고 싶을 때가 있는데 아래와 같이 간단히 설정 가능하다.
애용하는 방법이다.
Terminal에서 python 코드를 실행하는 경우
~$ CUDA_VISIBLE_DEVICES=0 python test.py
~$ CUDA_VISIBLE_DEVICES=1 python test.py
~$ CUDA_VISIBLE_DEVICES=2,3 python test.py
python 코드에서 GPU 지정하는 방법
import os os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
os.environ["CUDA_VISIBLE_DEVICES"]="0"
728x90
반응형
'Programming > Python' 카테고리의 다른 글
[Python] argparse(Argument Parser) 에서 boolean 값 받기 (0) | 2021.01.14 |
---|---|
[Python] pytube 이용하여 Youtube 영상 저장하기 (0) | 2020.12.28 |
[Python] 파일명 추출, 디렉토리 경로 추출 (3) | 2020.11.20 |
[Python] Python Schedule 모듈 이용하여 원하는 시간대에 프로그램 실행 시키기 (5) | 2020.11.02 |
[Python] Why time.time() gives 0.0? (0) | 2020.06.17 |