Commit ba6f6e44 authored by Julien Chaumond's avatar Julien Chaumond
Browse files

[ci] Re-enable torch GPU tests

parent 95249568
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: ...@@ -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: |
......
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