mirror of
https://github.com/explosion/spaCy.git
synced 2024-11-10 19:57:17 +03:00
Update CuPy extras (#12890)
* Add `cuda12x` for `cupy-cuda12x`. * Drop `cuda-autodetect` from quickstart, set default to `cuda11x` instead.
This commit is contained in:
parent
245e2ddc25
commit
c4e378df97
|
@ -111,6 +111,8 @@ cuda117 =
|
||||||
cupy-cuda117>=5.0.0b4,<13.0.0
|
cupy-cuda117>=5.0.0b4,<13.0.0
|
||||||
cuda11x =
|
cuda11x =
|
||||||
cupy-cuda11x>=11.0.0,<13.0.0
|
cupy-cuda11x>=11.0.0,<13.0.0
|
||||||
|
cuda12x =
|
||||||
|
cupy-cuda12x>=11.5.0,<13.0.0
|
||||||
cuda-autodetect =
|
cuda-autodetect =
|
||||||
cupy-wheel>=11.0.0,<13.0.0
|
cupy-wheel>=11.0.0,<13.0.0
|
||||||
apple =
|
apple =
|
||||||
|
|
|
@ -10,15 +10,19 @@ const DEFAULT_PLATFORM = 'x86'
|
||||||
const DEFAULT_MODELS = ['en']
|
const DEFAULT_MODELS = ['en']
|
||||||
const DEFAULT_OPT = 'efficiency'
|
const DEFAULT_OPT = 'efficiency'
|
||||||
const DEFAULT_HARDWARE = 'cpu'
|
const DEFAULT_HARDWARE = 'cpu'
|
||||||
const DEFAULT_CUDA = 'cuda-autodetect'
|
const DEFAULT_CUDA = 'cuda11x'
|
||||||
const CUDA = {
|
const CUDA = {
|
||||||
'8.0': 'cuda80',
|
'8.0': 'cuda80',
|
||||||
'9.0': 'cuda90',
|
'9.0': 'cuda90',
|
||||||
9.1: 'cuda91',
|
'9.1': 'cuda91',
|
||||||
9.2: 'cuda92',
|
'9.2': 'cuda92',
|
||||||
'10.0': 'cuda100',
|
'10.0': 'cuda100',
|
||||||
10.1: 'cuda101',
|
'10.1': 'cuda101',
|
||||||
'10.2, 11.0+': 'cuda-autodetect',
|
'10.2': 'cuda102',
|
||||||
|
'11.0': 'cuda110',
|
||||||
|
'11.1': 'cuda111',
|
||||||
|
'11.2-11.x': 'cuda11x',
|
||||||
|
'12.x': 'cuda12x',
|
||||||
}
|
}
|
||||||
const LANG_EXTRAS = ['ja'] // only for languages with models
|
const LANG_EXTRAS = ['ja'] // only for languages with models
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user