diff --git a/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb b/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
index 147ea316..6f4da7e9 100644
--- a/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
+++ b/how-to-use-azureml/explain-model/azure-integration/gpu-explanation/train-explain-model-gpu-tree-explainer.ipynb
@@ -269,7 +269,6 @@
         "printenv && \\\n",
         "echo \"which nvcc: \" && \\\n",
         "which nvcc && \\\n",
-        "pip install numpy==1.20.3 && \\\n",
         "pip install azureml-defaults && \\\n",
         "pip install azureml-telemetry && \\\n",
         "pip install azureml-interpret && \\\n",
@@ -281,7 +280,9 @@
         "mkdir build && \\\n",
         "python setup.py install --user && \\\n",
         "pip uninstall -y xgboost && \\\n",
-        "conda install py-xgboost==1.3.3 \\\n",
+        "conda install py-xgboost==1.3.3 && \\\n",
+        "pip uninstall -y numpy && \\\n",
+        "conda install numpy==1.20.3 \\\n",
         "\"\"\"\n",
         "\n",
         "env.python.user_managed_dependencies = True\n",