mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-20 01:27:06 -05:00
Merge pull request #1862 from Azure/release_update/Release-170
update samples from Release-170 as a part of SDK release
This commit is contained in:
@@ -256,7 +256,7 @@
|
|||||||
"\n",
|
"\n",
|
||||||
"# Note: we build shap at commit 690245 for Tesla K80 GPUs\n",
|
"# Note: we build shap at commit 690245 for Tesla K80 GPUs\n",
|
||||||
"env.docker.base_dockerfile = f\"\"\"\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",
|
"ENV PATH=\"/root/miniconda3/bin:${{PATH}}\"\n",
|
||||||
"ARG PATH=\"/root/miniconda3/bin:${{PATH}}\"\n",
|
"ARG PATH=\"/root/miniconda3/bin:${{PATH}}\"\n",
|
||||||
"RUN apt-get update && \\\n",
|
"RUN apt-get update && \\\n",
|
||||||
|
|||||||
@@ -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
|
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
|
ENV AZUREML_CONDA_ENVIRONMENT_PATH /azureml-envs/tensorflow-2.7
|
||||||
RUN apt-key del 7fa2af80
|
# Create conda environment
|
||||||
ENV distro ubuntu1804
|
RUN conda create -p $AZUREML_CONDA_ENVIRONMENT_PATH \
|
||||||
ENV arch x86_64
|
python=3.8 pip=20.2.4
|
||||||
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/$distro/$arch/3bf863cc.pub
|
|
||||||
|
|
||||||
|
# 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 \
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
||||||
python-opengl \
|
python-opengl \
|
||||||
rsync \
|
rsync \
|
||||||
@@ -17,15 +19,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
|
|||||||
rm -rf /var/lib/apt/lists/* && \
|
rm -rf /var/lib/apt/lists/* && \
|
||||||
rm -rf /usr/share/man/*
|
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 pip --version
|
||||||
RUN python --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[atari]==0.19.0
|
||||||
RUN pip install gym[accept-rom-license]==0.19.0
|
RUN pip install gym[accept-rom-license]==0.19.0
|
||||||
|
|
||||||
# Install pip dependencies
|
# Install pip dependencies for Tensorflow
|
||||||
RUN pip install 'matplotlib>=3.3,<3.4' \
|
RUN pip install 'matplotlib~=3.5.0' \
|
||||||
'psutil>=5.8,<5.9' \
|
'psutil~=5.8.0' \
|
||||||
'tqdm>=4.59,<4.60' \
|
'tqdm~=4.62.0' \
|
||||||
'pandas>=1.1,<1.2' \
|
'pandas~=1.3.0' \
|
||||||
'scipy>=1.5,<1.6' \
|
'scipy~=1.7.0' \
|
||||||
'numpy>=1.10,<1.20' \
|
'numpy~=1.21.0' \
|
||||||
'ipykernel~=6.0' \
|
'ipykernel~=6.0' \
|
||||||
'azureml-core==1.36.0.post2' \
|
'azureml-core==1.47.0' \
|
||||||
'azureml-defaults==1.36.0' \
|
'azureml-defaults==1.47.0' \
|
||||||
'azureml-mlflow==1.36.0' \
|
'azureml-mlflow==1.47.0' \
|
||||||
'azureml-telemetry==1.36.0' \
|
'azureml-telemetry==1.47.0' \
|
||||||
'tensorboard==2.4.0' \
|
'tensorboard~=2.7.0' \
|
||||||
'tensorflow-gpu==2.4.1' \
|
'tensorflow-gpu~=2.7.0' \
|
||||||
'tensorflow-datasets==4.3.0' \
|
'tensorflow-datasets~=4.5.0' \
|
||||||
'onnxruntime-gpu>=1.7,<1.8' \
|
'onnxruntime-gpu~=1.9.0' \
|
||||||
'horovod[tensorflow-gpu]==0.21.3'
|
'protobuf~=3.20' \
|
||||||
|
'horovod[tensorflow-gpu]~=0.23.0' \
|
||||||
|
'debugpy~=1.6.3'
|
||||||
|
|
||||||
RUN pip install --no-cache-dir \
|
RUN pip install --no-cache-dir \
|
||||||
azureml-defaults \
|
azureml-defaults \
|
||||||
|
|||||||
Reference in New Issue
Block a user