Extend to cupy 9.0.0 (#7914)

This commit is contained in:
Adriane Boyd 2021-04-28 10:18:24 +02:00 committed by GitHub
parent 8007d5c814
commit f4080983ea
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 12 additions and 12 deletions

View File

@ -41,7 +41,7 @@ steps:
displayName: "Install test requirements" displayName: "Install test requirements"
- script: | - script: |
${{ parameters.prefix }} python -m pip install -U cupy-cuda110 ${{ parameters.prefix }} python -m pip install -U cupy-cuda110 -f https://github.com/cupy/cupy/releases/v9.0.0
${{ parameters.prefix }} python -m pip install "torch==1.7.1+cu110" -f https://download.pytorch.org/whl/torch_stable.html ${{ parameters.prefix }} python -m pip install "torch==1.7.1+cu110" -f https://download.pytorch.org/whl/torch_stable.html
displayName: "Install GPU requirements" displayName: "Install GPU requirements"
condition: eq(${{ parameters.gpu }}, true) condition: eq(${{ parameters.gpu }}, true)

View File

@ -71,27 +71,27 @@ transformers =
ray = ray =
spacy_ray>=0.1.0,<1.0.0 spacy_ray>=0.1.0,<1.0.0
cuda = cuda =
cupy>=5.0.0b4,<9.0.0 cupy>=5.0.0b4,<10.0.0
cuda80 = cuda80 =
cupy-cuda80>=5.0.0b4,<9.0.0 cupy-cuda80>=5.0.0b4,<10.0.0
cuda90 = cuda90 =
cupy-cuda90>=5.0.0b4,<9.0.0 cupy-cuda90>=5.0.0b4,<10.0.0
cuda91 = cuda91 =
cupy-cuda91>=5.0.0b4,<9.0.0 cupy-cuda91>=5.0.0b4,<10.0.0
cuda92 = cuda92 =
cupy-cuda92>=5.0.0b4,<9.0.0 cupy-cuda92>=5.0.0b4,<10.0.0
cuda100 = cuda100 =
cupy-cuda100>=5.0.0b4,<9.0.0 cupy-cuda100>=5.0.0b4,<10.0.0
cuda101 = cuda101 =
cupy-cuda101>=5.0.0b4,<9.0.0 cupy-cuda101>=5.0.0b4,<10.0.0
cuda102 = cuda102 =
cupy-cuda102>=5.0.0b4,<9.0.0 cupy-cuda102>=5.0.0b4,<10.0.0
cuda110 = cuda110 =
cupy-cuda110>=5.0.0b4,<9.0.0 cupy-cuda110>=5.0.0b4,<10.0.0
cuda111 = cuda111 =
cupy-cuda111>=5.0.0b4,<9.0.0 cupy-cuda111>=5.0.0b4,<10.0.0
cuda112 = cuda112 =
cupy-cuda112>=5.0.0b4,<9.0.0 cupy-cuda112>=5.0.0b4,<10.0.0
# Language tokenizers with external dependencies # Language tokenizers with external dependencies
ja = ja =
sudachipy>=0.4.9 sudachipy>=0.4.9