fix rapids_sagemaker_hpo.ipynb (aws#3545)

hchings · Aug 19, 2022 · 92ce0e3 · 92ce0e3
1 parent 2db714c
commit 92ce0e3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/hyperparameter_tuning/rapids_bring_your_own/code/Dockerfile b/hyperparameter_tuning/rapids_bring_your_own/code/Dockerfile
@@ -8,14 +8,18 @@ ENV CV_FOLDS="3"
 # ensure printed output/log-messages retain correct order
 ENV PYTHONUNBUFFERED=True
 
+# delete expired nvidia keys and fetch new ones
+RUN apt-key del 7fa2af80
+RUN rm /etc/apt/sources.list.d/cuda.list
+RUN rm /etc/apt/sources.list.d/nvidia-ml.list
+RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb 
+
 # add sagemaker-training-toolkit [ requires build tools ], flask [ serving ], and dask-ml
 RUN apt-get update && apt-get install -y --no-install-recommends build-essential \ 
-    && source activate rapids && pip3 install sagemaker-training \
-    && conda install -c anaconda flask \
-    && conda install -c conda-forge dask-ml
+    && source activate rapids && pip3 install sagemaker-training dask-ml flask
 
 # path where SageMaker looks for code when container runs in the cloud
-ENV CLOUD_PATH="/opt/ml/code"
+ENV CLOUD_PATH "/opt/ml/code"
 
 # copy our latest [local] code into the container 
 COPY . $CLOUD_PATH

diff --git a/hyperparameter_tuning/rapids_bring_your_own/rapids_sagemaker_hpo.ipynb b/hyperparameter_tuning/rapids_bring_your_own/rapids_sagemaker_hpo.ipynb
@@ -704,14 +704,18 @@
     "# ensure printed output/log-messages retain correct order\n",
     "ENV PYTHONUNBUFFERED=True\n",
     "\n",
+    "# delete expired nvidia keys and fetch new ones\n",
+    "RUN apt-key del 7fa2af80\n",
+    "RUN rm /etc/apt/sources.list.d/cuda.list\n",
+    "RUN rm /etc/apt/sources.list.d/nvidia-ml.list\n",
+    "RUN wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-keyring_1.0-1_all.deb && dpkg -i cuda-keyring_1.0-1_all.deb \n",
+    "\n",
     "# add sagemaker-training-toolkit [ requires build tools ], flask [ serving ], and dask-ml\n",
     "RUN apt-get update && apt-get install -y --no-install-recommends build-essential \\ \n",
-    "    && source activate rapids && pip3 install sagemaker-training \\\n",
-    "    && conda install -c anaconda flask \\\n",
-    "    && conda install -c conda-forge dask-ml\n",
+    "    && source activate rapids && pip3 install sagemaker-training dask-ml flask\n",
     "\n",
     "# path where SageMaker looks for code when container runs in the cloud\n",
-    "ENV CLOUD_PATH=\"/opt/ml/code\"\n",
+    "ENV CLOUD_PATH \"/opt/ml/code\"\n",
     "\n",
     "# copy our latest [local] code into the container \n",
     "COPY . $CLOUD_PATH\n",