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)
on:
# push:
# branches:
# - master
push:
branches:
- master
# pull_request:
repository_dispatch:
......@@ -31,8 +31,8 @@ jobs:
- name: Install dependencies
run: |
source .env/bin/activate
pip install .[sklearn,tf,torch,testing]
pip uninstall -y tensorflow
pip install torch==1.4.0
pip install .[sklearn,testing]
- name: Are GPUs recognized by our DL frameworks
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