spaCy/enable_gpu.py at bbe4d3300a6c522ea4deb96d02dcac9da6a8ec5f - spaCy - Gitea

explosion/spaCy

mirror of https://github.com/explosion/spaCy.git synced 2024-12-30 20:06:30 +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()`