File tree 2 files changed +3
-5
lines changed 2 files changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -26,10 +26,11 @@ if [[ ${MATRIX_CHANNEL} != "release" ]]; then
26
26
fi
27
27
else
28
28
export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
29
+ poetry source add --priority=explicit pytorch " https://download.pytorch.org/whl/${MATRIX_DESIRED_CUDA} "
29
30
if [[ ${TORCH_ONLY} == ' true' ]]; then
30
31
poetry --quiet add torch
31
32
else
32
- poetry --quiet add torch torchaudio torchvision
33
+ poetry --quiet add --source pytorch torch torchaudio torchvision
33
34
fi
34
35
fi
35
36
Original file line number Diff line number Diff line change 68
68
eval "$(conda shell.bash hook)"
69
69
70
70
# Special case PyPi installation package. And Install of PyPi package via poetry
71
- if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" ]] && \
72
- ([[ ${MATRIX_GPU_ARCH_VERSION} == "12.1" && ${MATRIX_CHANNEL} != "release" ]] || \
73
- [[ ${MATRIX_GPU_ARCH_VERSION} == "11.7" && ${MATRIX_CHANNEL} == "release" ]]); then
71
+ if [[ ${MATRIX_PACKAGE_TYPE} == "manywheel" && ${MATRIX_GPU_ARCH_VERSION} == "12.1" ]]; then
74
72
source ./.github/scripts/validate_pipy.sh
75
- # temporary disable poetry check
76
73
source ./.github/scripts/validate_poetry.sh
77
74
fi
78
75
You can’t perform that action at this time.
0 commit comments