[Jupyter notebook] gpu 할당

-·2023년 10월 30일

0번(default) gpu말고 할당하기.

os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"  # Arrange GPU devices starting from 0
os.environ["CUDA_VISIBLE_DEVICES"]= "2,3"  # Set the GPUs 2 and 3 to use
profile
살아남은 자가 강한 것

0개의 댓글