spaCy/enable_gpu.py at 2a558a7cdcdaf817228154754c93b79642dc2bcd - spaCy - Gitea

explosion/spaCy

mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 19:39:13 +03:00

Adriane Boyd 36ecba224e

Set up GPU CI testing (#7293 )

* Set up CI for tests with GPU agent

* Update tests for enabled GPU

* Fix steps filename

* Add parallel build jobs as a setting

* Fix test requirements

* Fix install test requirements condition

* Fix pipeline models test

* Reset current ops in prefer/require testing

* Fix more tests

* Remove separate test_models test

* Fix regression 5551

* fix StaticVectors for GPU use

* fix vocab tests

* Fix regression test 5082

* Move azure steps to .github and reenable default pool jobs

* Consolidate/rename azure steps

Co-authored-by: svlandeg <sofie.vanlandeghem@gmail.com>

2021-04-22 14:58:29 +02:00

4 lines

45 B

Python

Raw Blame History

	`from spacy import require_gpu`

	`require_gpu()`