from __future__ import print_function
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
#os.environ['CUDA_VISIBLE_DEVICES'] = '-1' # Disable GPUfrom __future__ import print_function
import warnings
warnings.simplefilter(action='ignore', category=FutureWarning)
import os
os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
#os.environ['CUDA_VISIBLE_DEVICES'] = '-1' # Disable GPU