github.com/kubeflow/training-operator@v1.7.0/examples/pytorch/mnist/Dockerfile.ppc64le (about)

     1  FROM ppc64le/ubuntu:18.04
     2  WORKDIR /root/
     3  RUN apt-get update && apt-get -y install gcc g++ libjpeg-dev zlib1g-dev cmake
     4  RUN apt-get -y install python3 python3-pip git
     5  RUN pip3 install numpy pyyaml
     6  RUN git clone --recursive https://github.com/pytorch/pytorch && cd pytorch && python3 setup.py install
     7  RUN pip3 install torchvision tensorboardX==1.6.0
     8  WORKDIR /var
     9  ADD mnist.py /var
    10  
    11  ENTRYPOINT ["python3", "/var/mnist.py"]