[ci] Re-enable torch GPU tests

This commit is contained in:
Julien Chaumond
2020-05-12 00:05:36 +00:00
parent 9524956819
commit ba6f6e44a8

View File

@@ -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: |