Dockerfile

FROM pimachinelearning/raspi-python:3.9.16
ARG PYTORCH_VER=2.0.1
RUN apt-get update && apt-get install -y git libatlas3-base libgfortran5 ccache wget
RUN mkdir pytorch && wget https://github.com/pytorch/pytorch/releases/download/v${PYTORCH_VER}/pytorch-v${PYTORCH_VER}.tar.gz -O torch.tar.gz && tar -zxf torch.tar.gz --strip-components=1 -C pytorch || \
wget https://github.com/pytorch/pytorch/archive/refs/tags/v${PYTORCH_VER}.tar.gz -O torch.tar.gz && tar -zxf torch.tar.gz --strip-components=1 -C pytorch
COPY pip.conf /etc/pip.conf
RUN python3.9 -m pip install --upgrade pip
RUN python3.9 -m pip install cmake ninja
                # piwheels messed up 1.24.3 builds
RUN cd pytorch && python3.9 -m pip install numpy==1.24.2 && python3.9 -m pip install -r requirements.txt
COPY *.patch /pytorch/
COPY patch.sh /pytorch/
RUN cd pytorch && chmod +x patch.sh && ./patch.sh
CMD cd pytorch && MAX_JOBS=$(nproc --ignore=${IGNORE_CORES:-8}) PYTORCH_BUILD_NUMBER=0 USE_XNNPACK=OFF USE_CUDA=0 USE_CUDNN=0 USE_MKLDNN=0 USE_METAL=0 USE_NCCL=OFF USE_NNPACK=1 USE_QNNPACK=0 USE_PYTORCH_QNNPACK=0 USE_DISTRIBUTED=0 BUILD_TEST=0 CFLAGS="-march=armv8-a+crc+simd -mfpu=vfpv3-d16" python3.9 setup.py bdist_wheel