File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change 20
20
21
21
22
22
# Special case Pypi installation package, only applicable to linux nightly CUDA 11.7 builds, wheel package
23
- if [[ ${TARGET_OS} == ' linux' && ${MATRIX_GPU_ARCH_VERSION} == ' 11.7' && ${MATRIX_PACKAGE_TYPE} == ' manywheel' ]]; then
23
+ if [[ ${TARGET_OS} == ' linux' && ${MATRIX_GPU_ARCH_VERSION} == ' 11.7' && ${MATRIX_PACKAGE_TYPE} == ' manywheel' && ${MATRIX_CHANNEL} != ' nightly ' ]]; then
24
24
conda create -yp ${ENV_NAME} _pypi python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
25
25
INSTALLATION_PYPI=${MATRIX_INSTALLATION/ " cu117" / " cu117_pypi_cudnn" }
26
26
INSTALLATION_PYPI=${INSTALLATION_PYPI/ " torchvision torchaudio" / " " }
35
35
conda create -y -n ${ENV_NAME} python=${MATRIX_PYTHON_VERSION} numpy ffmpeg
36
36
conda activate ${ENV_NAME}
37
37
INSTALLATION=${MATRIX_INSTALLATION/ " conda install" / " conda install -y" }
38
- INSTALLATION=${INSTALLATION/ " extra-index-url" / " index-url" }
39
38
eval $INSTALLATION
40
39
41
40
if [[ ${TARGET_OS} == ' linux' ]]; then
Original file line number Diff line number Diff line change 36
36
def check_version (package : str ) -> None :
37
37
# only makes sense to check nightly package where dates are known
38
38
if channel == "nightly" :
39
- check_nightly_binaries_date (options . package )
39
+ check_nightly_binaries_date (package )
40
40
else :
41
41
if not torch .__version__ .startswith (stable_version ):
42
42
raise RuntimeError (
You can’t perform that action at this time.
0 commit comments