explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 09:56:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Remove cuda extras for non-linux arm in install widget (#10796)

Browse Source 
* Remove cuda extras for non-linux arm platforms in install widget
* Extend cuda versions install widget
* Update GPU install docs to clarify cuda
This commit is contained in:
Adriane Boyd 2022-05-20 09:57:41 +02:00 committed by GitHub
parent 46982cf694
commit a82ec56aae
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
website/docs/usage/index.md
View File

@ -129,15 +129,14 @@ machine learning library, [Thinc](https://thinc.ai). For GPU support, we've been
grateful to use the work of Chainer's [CuPy](https://cupy.chainer.org) module, grateful to use the work of Chainer's [CuPy](https://cupy.chainer.org) module,
which provides a numpy-compatible interface for GPU arrays. which provides a numpy-compatible interface for GPU arrays.
spaCy can be installed on GPU by specifying `spacy[cuda]`, `spacy[cuda90]`, spaCy can be installed for a CUDA-compatible GPU by specifying `spacy[cuda]`,
`spacy[cuda91]`, `spacy[cuda92]`, `spacy[cuda100]`, `spacy[cuda101]`, `spacy[cuda102]`, `spacy[cuda112]`, `spacy[cuda113]`, etc. If you know your
`spacy[cuda102]`, `spacy[cuda110]`, `spacy[cuda111]` or `spacy[cuda112]`. If you CUDA version, using the more explicit specifier allows CuPy to be installed via
know your cuda version, using the more explicit specifier allows cupy to be wheel, saving some compilation time. The specifiers should install
installed via wheel, saving some compilation time. The specifiers should install
[`cupy`](https://cupy.chainer.org). [`cupy`](https://cupy.chainer.org).
```bash ```bash
$ pip install -U %%SPACY_PKG_NAME[cuda92]%%SPACY_PKG_FLAGS $ pip install -U %%SPACY_PKG_NAME[cuda113]%%SPACY_PKG_FLAGS
``` ```
Once you have a GPU-enabled installation, the best way to activate it is to call Once you have a GPU-enabled installation, the best way to activate it is to call

4
website/src/widgets/quickstart-install.js
View File

@ -23,6 +23,8 @@ const CUDA = {
'11.2': 'cuda112', '11.2': 'cuda112',
'11.3': 'cuda113', '11.3': 'cuda113',
'11.4': 'cuda114', '11.4': 'cuda114',
'11.5': 'cuda115',
'11.6': 'cuda116',
} }
const LANG_EXTRAS = ['ja'] // only for languages with models const LANG_EXTRAS = ['ja'] // only for languages with models
@ -48,7 +50,7 @@ const QuickstartInstall = ({ id, title }) => {
const modelExtras = train ? selectedModels.filter(m => LANG_EXTRAS.includes(m)) : [] const modelExtras = train ? selectedModels.filter(m => LANG_EXTRAS.includes(m)) : []
const apple = os === 'mac' && platform === 'arm' const apple = os === 'mac' && platform === 'arm'
const pipExtras = [ const pipExtras = [
hardware === 'gpu' && cuda, (hardware === 'gpu' && (platform !== 'arm' || os === 'linux')) && cuda,
train && 'transformers', train && 'transformers',
train && 'lookups', train && 'lookups',
apple && 'apple', apple && 'apple',