From 3a02a27f1e46347dcb731d67cc6c8d2f9d010e8b Mon Sep 17 00:00:00 2001
From: amlrelsa-ms <amlrelsa@microsoft.com>
Date: Tue, 6 Dec 2022 03:22:18 +0000
Subject: [PATCH] update samples from Release-170 as a part of  SDK release

---
 ...ain-explain-model-gpu-tree-explainer.ipynb |  2 +-
 .../docker/Dockerfile-gpu                     | 59 +++++++++----------
 2 files changed, 28 insertions(+), 33 deletions(-)

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 729d5918..cb629f2b 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
@@ -256,7 +256,7 @@
         "\n",
         "# Note: we build shap at commit 690245 for Tesla K80 GPUs\n",
         "env.docker.base_dockerfile = f\"\"\"\n",
-        "FROM nvidia/cuda:10.2-devel-ubuntu20.04\n",
+        "FROM nvidia/cuda:10.2-devel-ubuntu18.04\n",
         "ENV PATH=\"/root/miniconda3/bin:${{PATH}}\"\n",
         "ARG PATH=\"/root/miniconda3/bin:${{PATH}}\"\n",
         "RUN apt-get update && \\\n",
diff --git a/how-to-use-azureml/reinforcement-learning/atari-on-distributed-compute/docker/Dockerfile-gpu b/how-to-use-azureml/reinforcement-learning/atari-on-distributed-compute/docker/Dockerfile-gpu
index 93ab770c..01dcdc93 100644
--- a/how-to-use-azureml/reinforcement-learning/atari-on-distributed-compute/docker/Dockerfile-gpu
+++ b/how-to-use-azureml/reinforcement-learning/atari-on-distributed-compute/docker/Dockerfile-gpu
@@ -1,14 +1,16 @@
-FROM mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.1-cudnn8-ubuntu20.04
+FROM mcr.microsoft.com/azureml/openmpi4.1.0-cuda11.2-cudnn8-ubuntu20.04:20221010.v1
 
 USER root
-RUN conda install -c anaconda python=3.7
 
-# CUDA repository key rotation: https://forums.developer.nvidia.com/t/notice-cuda-linux-repository-key-rotation/212771
-RUN apt-key del 7fa2af80
-ENV distro ubuntu1804
-ENV arch x86_64
-RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/3bf863cc.pub
+ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/tensorflow-2.7
+# Create conda environment
+RUN conda create -p $AZUREML_CONDA_ENVIRONMENT_PATH \
+    python=3.8 pip=20.2.4
 
+# Prepend path to AzureML conda environment
+ENV PATH $AZUREML_CONDA_ENVIRONMENT_PATH/bin:$PATH
+
+# Install necessary packages to support videos in rllib/gym
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python-opengl \
     rsync \
@@ -17,15 +19,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     rm -rf /var/lib/apt/lists/* && \
     rm -rf /usr/share/man/*
 
-ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/tensorflow-2.4
-
-# Create conda environment
-RUN conda create -p $AZUREML_CONDA_ENVIRONMENT_PATH \
-    python=3.7 pip=20.2.4
-
-# Prepend path to AzureML conda environment
-ENV PATH $AZUREML_CONDA_ENVIRONMENT_PATH/bin:$PATH
-
 RUN pip --version
 RUN python --version
 
@@ -36,23 +29,25 @@ RUN pip install ray==0.8.7
 RUN pip install gym[atari]==0.19.0
 RUN pip install gym[accept-rom-license]==0.19.0
 
-# Install pip dependencies
-RUN pip install 'matplotlib>=3.3,<3.4' \
-                'psutil>=5.8,<5.9' \
-                'tqdm>=4.59,<4.60' \
-                'pandas>=1.1,<1.2' \
-                'scipy>=1.5,<1.6' \
-                'numpy>=1.10,<1.20' \
+# Install pip dependencies for Tensorflow
+RUN pip install 'matplotlib~=3.5.0' \
+                'psutil~=5.8.0' \
+                'tqdm~=4.62.0' \
+                'pandas~=1.3.0' \
+                'scipy~=1.7.0' \
+                'numpy~=1.21.0' \
                 'ipykernel~=6.0' \
-                'azureml-core==1.36.0.post2' \
-                'azureml-defaults==1.36.0' \
-                'azureml-mlflow==1.36.0' \
-                'azureml-telemetry==1.36.0' \
-                'tensorboard==2.4.0' \
-                'tensorflow-gpu==2.4.1' \
-                'tensorflow-datasets==4.3.0' \
-                'onnxruntime-gpu>=1.7,<1.8' \
-                'horovod[tensorflow-gpu]==0.21.3'
+                'azureml-core==1.47.0' \
+                'azureml-defaults==1.47.0' \
+                'azureml-mlflow==1.47.0' \
+                'azureml-telemetry==1.47.0' \
+                'tensorboard~=2.7.0' \
+                'tensorflow-gpu~=2.7.0' \
+                'tensorflow-datasets~=4.5.0' \
+                'onnxruntime-gpu~=1.9.0' \
+                'protobuf~=3.20' \
+                'horovod[tensorflow-gpu]~=0.23.0' \
+                'debugpy~=1.6.3'
 
 RUN pip install --no-cache-dir \
     azureml-defaults \