Update CuPy extras (#12890)

* Add `cuda12x` for `cupy-cuda12x`.
* Drop `cuda-autodetect` from quickstart, set default to `cuda11x`
instead.
This commit is contained in:
Adriane Boyd 2023-08-08 12:58:28 +02:00 committed by GitHub
parent 245e2ddc25
commit c4e378df97
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 11 additions and 5 deletions

View File

@ -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 =

View File

@ -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