ci: fix install lite on GPUs (#15375)
This commit is contained in:
parent
773cb3e8c8
commit
e31a67541b
|
@ -63,17 +63,13 @@ jobs:
|
|||
|
||||
- bash: |
|
||||
set -e
|
||||
PYTORCH_VERSION=$(python -c "import torch; print(torch.__version__.split('+')[0])")
|
||||
python ./requirements/pytorch/adjust-versions.py requirements/lite/base.txt ${PYTORCH_VERSION}
|
||||
|
||||
CUDA_VERSION_MM=$(python -c "import torch ; print(''.join(map(str, torch.version.cuda.split('.')[:2])))")
|
||||
pip install -e .[dev,strategies] --find-links https://download.pytorch.org/whl/cu${CUDA_VERSION_MM}/torch_stable.html
|
||||
|
||||
pip list
|
||||
env:
|
||||
PACKAGE_NAME: pytorch
|
||||
FREEZE_REQUIREMENTS: 1
|
||||
displayName: 'Install dependencies'
|
||||
PACKAGE_NAME: "lite"
|
||||
FREEZE_REQUIREMENTS: "1"
|
||||
displayName: 'Install package & dependencies'
|
||||
|
||||
- bash: |
|
||||
set -e
|
||||
|
|
Loading…
Reference in New Issue