update Dockerfile to use v1.5.1 (#132)

* update Dockerfile to use v1.5.1

Author: ashokei

docker/Dockerfile

@@ -10,22 +10,18 @@
 #           https://docs.docker.com/develop/develop-images/build_enhancements/
 
 ARG BASE_IMAGE=ubuntu:20.04
-ARG PYTHON_VERSION=3.8
-
 FROM ${BASE_IMAGE} AS dev-base
 RUN --mount=type=cache,id=apt-dev,target=/var/cache/apt \
     apt-get update && \
     DEBIAN_FRONTEND=noninteractive apt-get install --no-install-recommends -y \
-    fakeroot \
     ca-certificates \
-    dpkg-dev \
-    g++ \
     git \
     curl \
     vim \
     build-essential \
     ccache \
     libjemalloc-dev \
+    numactl \
     cmake \
     libjpeg-dev \
     libpng-dev \
@@ -35,6 +31,7 @@ RUN mkdir /opt/ccache && ccache --set-config=cache_dir=/opt/ccache
 ENV PATH /opt/conda/bin:$PATH
 
 FROM dev-base as conda
+ARG PYTHON_VERSION=3.8
 RUN curl -fsSL -v -o ~/miniconda.sh -O  https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh  && \
     chmod +x ~/miniconda.sh && \
     ~/miniconda.sh -b -p /opt/conda && \
@@ -45,22 +42,28 @@ RUN curl -fsSL -v -o ~/miniconda.sh -O  https://repo.anaconda.com/miniconda/Mini
 FROM dev-base AS build
 COPY --from=conda /opt/conda /opt/conda
 ARG TORCHVISION_VERSION=0.6
-RUN pip install torchvision==${TORCHVISION_VERSION}+cpu --no-deps -f https://download.pytorch.org/whl/torch_stable.html && \
-    pip install lark-parser hypothesis && git clone https://github.com/intel/intel-extension-for-pytorch \
-    && cd intel-extension-for-pytorch && git submodule sync && git submodule update --init --recursive \
-    && git clone https://github.com/pytorch/pytorch \
-    && cd pytorch && git checkout v1.5.0-rc3 && git submodule sync && git submodule update --init --recursive \
-    && git apply ../torch_patches/dpcpp-v1.5-rc3.patch \
-    && USE_MKLDNN=1 USE_CUDA=0 USE_CUDNN=0 CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" python setup.py install \
-    && cd .. && python setup.py install && rm -rf *
+RUN --mount=type=cache,target=/opt/ccache \
+    pip install torchvision==${TORCHVISION_VERSION}+cpu --no-deps \
+    -f https://download.pytorch.org/whl/torch_stable.html && \
+    pip install lark-parser hypothesis && \
+    git clone https://github.com/intel/intel-extension-for-pytorch && \
+    cd intel-extension-for-pytorch && git submodule sync && \
+    git submodule update --init --recursive && \
+    git clone https://github.com/pytorch/pytorch && \
+    cd pytorch && git checkout v1.5.1 && git submodule sync && \
+    git submodule update --init --recursive && \
+    git apply ../torch_patches/dpcpp-v1.5.1.patch && \
+    USE_MKLDNN=1 USE_CUDA=0 USE_NNPACK=0 USE_CUDNN=0 \
+    CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" pip install -v . && \
+    cd .. && pip install -v . && rm -rf *
 
 FROM dev-base as dev
 COPY --from=build /opt/conda /opt/conda
 ARG OMP_NUM_THREADS=1
 ENV OMP_NUM_THREADS ${OMP_NUM_THREADS}
 ARG KMP_BLOCKTIME=1
 ENV KMP_BLOCKTIME ${KMP_BLOCKTIME} 
-ARG KMP_AFFINITY="granularity=fine,compact,1,0"
-ENV KMP_AFFINITY ${KMP_AFFINITY}
+ARG KMP_HW_SUBSET=1T
+ENV KMP_HW_SUBSET ${KMP_HW_SUBSET}
 ENV MALLOC_CONF "oversize_threshold:1,background_thread:true,metadata_thp:auto,dirty_decay_ms:-1,muzzy_decay_ms:-1"
 ENV LD_PRELOAD "/opt/conda/lib/libiomp5.so /usr/lib/x86_64-linux-gnu/libjemalloc.so"