diff --git a/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb b/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb index 0414ba68a8..0b6b23ff74 100644 --- a/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb +++ b/tutorials/e2e-distributed-pytorch-image/e2e-object-classification-distributed-pytorch.ipynb @@ -373,7 +373,8 @@ " code=\"./src/pytorch_dl_train/\",\n", " # describe the command to run the python script, with all its parameters\n", " # use the syntax below to inject parameter values from code\n", - " command=\"\"\"python train.py \\\n", + " command=\"\"\"python -m pip install --no-cache-dir --no-deps \"protobuf==3.20.3\" && \\\n", + " python train.py \\\n", " --train_images ${{inputs.train_images}} \\\n", " --valid_images ${{inputs.valid_images}} \\\n", " --batch_size ${{inputs.batch_size}} \\\n",