cuda에서 gpu 할당

특정 gpu에만 gpu 할당하는 방법

Bash에서 실행 시 반영하려면

CUDA_VISIBLE_DEVICES=1 [command]

CUDA_VISIBLE_DEVICES=1,2,3 [command]

CUDA_VISIBLE_DEVICES 파이썬에서 코딩하기

출처

https://stackoverflow.com/questions/37893755/tensorflow-set-cuda-visible-devices-within-jupyter/37901914

import os

os.environ["CUDA_DEVICE_ORDER"]    = "PCI_BUS_ID"

os.environ["CUDA_VISIBLE_DEVICES"] = "0"