hammacktony
11/1/2019 - 1:56 PM

GPU Cheat

GPU Cheat

import os

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