Pin torch to a specific version in the CI.

This temporarily works around the CMake error:

```
CMake Error in frontends/pytorch/csrc/CMakeLists.txt:
  Imported target "torch" includes non-existent path

    "/pytorch/torch/lib"

  in its INTERFACE_INCLUDE_DIRECTORIES.
```
pull/234/head
Sean Silva 2021-06-22 13:10:24 -07:00
parent e6adecac83
commit f7ebd870f6
1 changed files with 1 additions and 1 deletions

View File

@ -20,7 +20,7 @@ jobs:
python3 -m pip install pytest pybind11 numpy python3 -m pip install pytest pybind11 numpy
- name: Install pytorch_nightly depends - name: Install pytorch_nightly depends
run: | run: |
python3 -m pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html python3 -m pip install --pre 'torch==1.10.0.dev20210617+cpu' 'torchvision==0.11.0.dev20210617+cpu' -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
- name: Get LLVM Hash - name: Get LLVM Hash
id: get-llvm-hash id: get-llvm-hash
run: echo "::set-output name=hash::$(git submodule status)" run: echo "::set-output name=hash::$(git submodule status)"