Dockerfile_MindSpore

# Download base image from NVIDIA's Docker Hub
FROM nvidia/cuda:11.6.2-cudnn8-runtime-ubuntu20.04
LABEL maintainer="Baochun Li"

ADD ./.bashrc /root/
COPY ./requirements.txt /root/
WORKDIR /root/plato

RUN apt-get update \
    && apt-get install -y wget \
    && apt-get install -y vim \
    && apt-get install -y net-tools \
    && apt-get install -y git \
    && mkdir -p ~/miniconda3 \
    && wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-aarch64.sh -O ~/miniconda3/miniconda.sh \
    && bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3 \
    && rm -rf ~/miniconda3/miniconda.sh \
    && ~/miniconda3/bin/conda update -n base -c defaults conda \
    && ~/miniconda3/bin/conda init bash \
    && ~/miniconda3/bin/conda create -n plato_gpu -c conda-forge python=3.9 \
    && ~/miniconda3/bin/conda install mindspore=2.0.0 -c mindspore -c conda-forge -n plato_gpu -y \
    && ~/miniconda3/envs/plato_gpu/bin/pip install -r ~/requirements.txt \
    && ~/miniconda3/envs/plato_gpu/bin/pip install plato-learn \
    && ~/miniconda3/bin/conda create -n plato_cpu -c conda-forge python=3.9 \
    && ~/miniconda3/bin/conda install mindspore=2.0.0 -c mindspore -c conda-forge -n plato_cpu -y \
    && ~/miniconda3/envs/plato_cpu/bin/pip install -r ~/requirements.txt \
    && ~/miniconda3/envs/plato_cpu/bin/pip install Pillow==9.5.0 \
    && ~/miniconda3/envs/plato_cpu/bin/pip install plato-learn 

RUN rm /root/requirements.txt