Skip to content

Commit 9afd0df

Browse files
committed
Update the docker to PT1.7
1 parent 64ea758 commit 9afd0df

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docker/Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,9 +50,9 @@ RUN --mount=type=cache,target=/opt/ccache \
5050
cd intel-extension-for-pytorch && git submodule sync && \
5151
git submodule update --init --recursive && \
5252
git clone https://github.com/pytorch/pytorch && \
53-
cd pytorch && git checkout v1.5.1 && git submodule sync && \
53+
cd pytorch && git checkout v1.7.0 && git submodule sync && \
5454
git submodule update --init --recursive && \
55-
git apply ../torch_patches/dpcpp-v1.5.1.patch && \
55+
git apply ../torch_patches/xpu-1.7.patch && \
5656
USE_MKLDNN=1 USE_CUDA=0 USE_NNPACK=0 USE_CUDNN=0 \
5757
CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" pip install -v . && \
5858
cd .. && pip install -v . && rm -rf *

0 commit comments

Comments
 (0)