import os os.environ["CUDA_VISIBLE_DEVICES"] = "1" from . import *