Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

fix rapids_sagemaker_hpo.ipynb #3545

Merged
merged 4 commits into from
Aug 19, 2022
Merged

fix rapids_sagemaker_hpo.ipynb #3545

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
1cacfce
fix rapids_sagemaker_hpo.ipynb
atqy Aug 16, 2022
358b1ba
Merge branch 'main' into atqy-fix-byoc-rapids-sagemaker-hpo
atqy Aug 16, 2022
da1c102
Merge branch 'main' into atqy-fix-byoc-rapids-sagemaker-hpo
atqy Aug 16, 2022
51aa14b
Merge branch 'main' into atqy-fix-byoc-rapids-sagemaker-hpo
atqy Aug 18, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 8 additions & 4 deletions hyperparameter_tuning/rapids_bring_your_own/code/Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
12 changes: 8 additions & 4 deletions hyperparameter_tuning/rapids_bring_your_own/rapids_sagemaker_hpo.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down