From ac36e4bfa691900a052d45251e3eb901f5062714 Mon Sep 17 00:00:00 2001 From: Sam Daulton Date: Thu, 11 Apr 2024 09:16:34 -0700 Subject: [PATCH] fix --- .github/workflows/reusable_tutorials.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/reusable_tutorials.yml b/.github/workflows/reusable_tutorials.yml index 29c5da043d..33177dc703 100644 --- a/.github/workflows/reusable_tutorials.yml +++ b/.github/workflows/reusable_tutorials.yml @@ -50,6 +50,7 @@ jobs: - if: ${{ !inputs.use_stable_pytorch_gpytorch }} name: Install latest PyTorch & GPyTorch run: | + pip install "numpy==1.26.4" pip install --pre torch torchvision -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html pip install git+https://github.com/cornellius-gp/linear_operator.git pip install git+https://github.com/cornellius-gp/gpytorch.git @@ -63,7 +64,7 @@ jobs: min_torch_version=$(grep '\btorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g') min_gpytorch_version=$(grep '\bgpytorch[>=]=' ${req_txt} | sed 's/[^0-9.]//g') min_linear_operator_version=$(grep '\blinear_operator[>=]=' ${req_txt} | sed 's/[^0-9.]//g') - pip install "numpy==${max_numpy_version}" "torch==${min_torch_version}" "gpytorch==${min_gpytorch_version}" "linear_operator==${min_linear_operator_version}" torchvision + pip install "numpy==1.26.4" "torch==${min_torch_version}" "gpytorch==${min_gpytorch_version}" "linear_operator==${min_linear_operator_version}" torchvision - name: Install BoTorch with tutorials dependencies env: ALLOW_LATEST_GPYTORCH_LINOP: true