Unverified Commit 1f8e8265 authored by Sam Shleifer's avatar Sam Shleifer Committed by GitHub
Browse files

[CI] Self-scheduled runner also pins torch (#6332)

parent 1b8a7ffc
...@@ -32,7 +32,8 @@ jobs: ...@@ -32,7 +32,8 @@ jobs:
run: | run: |
source .env/bin/activate source .env/bin/activate
pip install --upgrade pip 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 - name: Are GPUs recognized by our DL frameworks
run: | run: |
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment