spaCy/enable_gpu.py at c1caa47aa74bd4ddd820c0976aa8e6410bd00984 - spaCy - Gitea

explosion/spaCy

mirror of https://github.com/explosion/spaCy.git synced 2024-11-14 05:37:03 +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()`