Fix device recognition issue when NV GPU is present
@@ -51,6 +51,8 @@ class TensorFlowAdaptor(Adaptor):
def __init__(self, framework_specific_info):
super().__init__(framework_specific_info)
+ os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"
+ os.environ['CUDA_VISIBLE_DEVICES'] = "-1"
self.quantize_config = {'op_wise_config': {}}
self.framework_specific_info = framework_specific_info
self.approach = deep_get(self.framework_specific_info, 'approach', False)
