[ci] Re-enable torch GPU tests
This commit is contained in:
10
.github/workflows/self-push.yml
vendored
10
.github/workflows/self-push.yml
vendored
@@ -1,9 +1,9 @@
|
|||||||
name: Self-hosted runner (push)
|
name: Self-hosted runner (push)
|
||||||
|
|
||||||
on:
|
on:
|
||||||
# push:
|
push:
|
||||||
# branches:
|
branches:
|
||||||
# - master
|
- master
|
||||||
# pull_request:
|
# pull_request:
|
||||||
repository_dispatch:
|
repository_dispatch:
|
||||||
|
|
||||||
@@ -31,8 +31,8 @@ jobs:
|
|||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
source .env/bin/activate
|
source .env/bin/activate
|
||||||
pip install .[sklearn,tf,torch,testing]
|
pip install torch==1.4.0
|
||||||
pip uninstall -y tensorflow
|
pip install .[sklearn,testing]
|
||||||
|
|
||||||
- name: Are GPUs recognized by our DL frameworks
|
- name: Are GPUs recognized by our DL frameworks
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
Reference in New Issue
Block a user