add neural_compressor version check (#1595)

* add neural_compressor version check

* modify error message

Author: jiayisunx

examples/cpu/features/int8_recipe_tuning/imagenet_autotune.py

@@ -126,6 +126,8 @@ def eval_func(model):
       images = images.contiguous(memory_format=torch.channels_last)
       traced_model = torch.jit.trace(converted_model, images)
       traced_model = torch.jit.freeze(traced_model)
+      traced_model(images)
+      traced_model(images)
       break
 
   eval_func(traced_model)
@@ -166,7 +168,7 @@ def main(args):
                  ' (default: resnet18)')
   parser.add_argument('-j', '--workers', default=4, type=int, metavar='N',
             help='number of data loading workers (default: 4)')
-  parser.add_argument('-b', '--batch-size', default=56, type=int,
+  parser.add_argument('-b', '--batch-size', default=256, type=int,
             metavar='N',
             help='mini-batch size (default: 256), this is the total '
                  'batch size of all GPUs on the current node when '

intel_extension_for_pytorch/quantization/_autotune.py

@@ -24,11 +24,14 @@ def autotune(prepared_model, calib_dataloader, eval_func, sampling_sizes=[100],
     Returns:
         FP32 tuned model (torch.nn.Module)
     """
+    neural_compressor_version = '1.14.1'
     try:
         import neural_compressor
+        if neural_compressor.__version__ != neural_compressor_version:
+            raise RuntimeError("Please install Intel® Neural Compressor with version {} while the current version of Intel® Neural Compressor is {}.".format(neural_compressor_version, neural_compressor.__version__))
     except ImportError:
         try:
-            subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'neural_compressor==1.14.1'])
+            subprocess.check_call([sys.executable, '-m', 'pip', 'install', 'neural_compressor=={}'.format(neural_compressor_version)])
             import neural_compressor
         except:
             assert False, "Unable to import neural_compressor from the local environment."