From 1f8e8265188de8b76f5c28539056d6eb772e4e0f Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Fri, 7 Aug 2020 18:40:21 -0400 Subject: [PATCH] [CI] Self-scheduled runner also pins torch (#6332) --- .github/workflows/self-scheduled.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/self-scheduled.yml b/.github/workflows/self-scheduled.yml index a61a9b5807..afd6c6b01b 100644 --- a/.github/workflows/self-scheduled.yml +++ b/.github/workflows/self-scheduled.yml @@ -32,7 +32,8 @@ jobs: run: | source .env/bin/activate pip install --upgrade pip - pip install .[sklearn,torch,testing] + pip install torch!=1.6.0 --no-cache-dir + pip install .[sklearn,testing] - name: Are GPUs recognized by our DL frameworks run: |