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"

 

 

 

참고자료 : dongjinlee.tistory.com/entry/%EC%84%A0%ED%83%9D%ED%95%9C-GPU%EC%97%90%EB%A7%8C-%EB%A9%94%EB%AA%A8%EB%A6%AC-%ED%95%A0%EB%8B%B9%ED%95%98%EB%8A%94-%EB%B0%A9%EB%B2%95

 

선택한 GPU에만 메모리 할당하는 방법

출처: https://www.tensorflow.org/tutorials/using_gpu Issue: multi-GPU 시스템에서 tensorflow를 실행할 경우 하나의 GPU에서 코드가 실행되게 프로그래밍을 해도 모든 GPU에 메모리를 할당한다. 이는 위의..

dongjinlee.tistory.com

 

728x90
반응형