Merge branch 'main' into patch-1

Author: Svetlana Karslioglu

.jenkins/validate_tutorials_built.py

@@ -49,7 +49,6 @@
     "recipes_source/recipes/Captum_Recipe",
     "intermediate_source/flask_rest_api_tutorial",
     "intermediate_source/text_to_speech_with_torchaudio",
-    "intermediate_source/ax_multiobjective_nas_tutorial",
     "intermediate_source/tensorboard_profiler_tutorial" # reenable after 2.0 release.
 ]
 

intermediate_source/mnist_train_nas.py

@@ -44,7 +44,6 @@ def parse_args():
 args = parse_args()
 
 PATH_DATASETS = os.environ.get("PATH_DATASETS", ".")
-AVAIL_GPUS = min(1, torch.cuda.device_count())
 
 
 class MnistModel(LightningModule):
@@ -136,7 +135,6 @@ def run_training_job():
     # Initialize a trainer (don't log anything since things get so slow...)
     trainer = Trainer(
         logger=False,
-        gpus=AVAIL_GPUS,
         max_epochs=args.epochs,
         enable_progress_bar=False,
         deterministic=True,  # Do we want a bit of noise?