From 10a2829dd829a689bd03388e504a6826a8f9399a Mon Sep 17 00:00:00 2001 From: Chen Wang Date: Sat, 21 Oct 2023 23:31:54 -0400 Subject: [PATCH] Update torch to 2.* (#296) * Update runtime.txt Signed-off-by: Chen Wang * Update config.yml Signed-off-by: Chen Wang --------- Signed-off-by: Chen Wang --- .circleci/config.yml | 6 +++--- requirements/runtime.txt | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.circleci/config.yml b/.circleci/config.yml index 20eb6e1ec..aaba16902 100644 --- a/.circleci/config.yml +++ b/.circleci/config.yml @@ -151,15 +151,15 @@ workflows: jobs: - linux_build_wheel: cuda_version: "11.7" - torch_version: "2.0.*" + torch_version: "2.*" python_version: "3.9" - build_docs: requires: [linux_build_wheel] cuda_version: "11.7" - torch_version: "2.0.*" + torch_version: "2.*" python_version: "3.9" - run_tests: requires: [linux_build_wheel] cuda_version: "11.7" - torch_version: "2.0.*" + torch_version: "2.*" python_version: "3.9" diff --git a/requirements/runtime.txt b/requirements/runtime.txt index 9562e1a46..09b07dac0 100644 --- a/requirements/runtime.txt +++ b/requirements/runtime.txt @@ -1,2 +1,2 @@ pytest==7.* -torch==2.0.* +torch==2.*