got this , will have to set the os.environ["CUDA_VISIBLE_DEVICES"] = "MIG-53d2748f-2b8......"