mirror of
https://github.com/Azure/MachineLearningNotebooks.git
synced 2025-12-23 02:52:39 -05:00
17 lines
732 B
Docker
17 lines
732 B
Docker
# From https://github.com/microsoft/AzureML-BERT/blob/master/finetune/PyTorch/dockerfile
|
|
|
|
FROM mcr.microsoft.com/azureml/base-gpu:openmpi3.1.2-cuda10.1-cudnn7-ubuntu18.04
|
|
|
|
RUN apt update && apt install git -y && rm -rf /var/lib/apt/lists/*
|
|
|
|
RUN /opt/miniconda/bin/conda update -n base -c defaults conda
|
|
RUN /opt/miniconda/bin/conda install -y cython=0.29.15 numpy=1.18.1
|
|
RUN /opt/miniconda/bin/conda install -y pytorch=1.4 torchvision=0.5.0 -c pytorch
|
|
|
|
# Install cocoapi, required for drawing bounding boxes
|
|
RUN git clone https://github.com/cocodataset/cocoapi.git && cd cocoapi/PythonAPI && python setup.py build_ext install
|
|
|
|
RUN pip install azureml-defaults
|
|
RUN pip install "azureml-dataprep[fuse]"
|
|
RUN pip install pandas pyarrow
|