“Setel perangkat Python yang terlihat cuda” Kode Jawaban

Setel perangkat Python yang terlihat cuda

import os
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"   # see issue #152
os.environ["CUDA_VISIBLE_DEVICES"]="0"